tinygrad
tinygrad copied to clipboard
Tensor method support dim <-> axis and keepdim <-> keepdims
default is dim and keepdim like torch, and axis / keepdims work