Image
Meaning
✅
Done
🔵
In progress
❌
Todo
Numpy-like operations over a raw-tensor object
Functionality
CPU
GPU
Comments
zeros
✅
✅
ones
✅
✅
arange
✅
✅
range
✅
✅
linspace
✅
✅
logspace
✅
✅
eye
✅
✅
full
✅
✅
Indexing, Slicing, Joining, Mutating Ops
Functionality
CPU
GPU
Comments
cat
❌
❌
chunk
❌
❌
gather
❌
❌
nonzero
❌
❌
reshape
✅
✅
split
❌
❌
squeeze
❌
❌
stack
❌
❌
transpose
✅
✅
unsqueeze
❌
❌
where
❌
❌
get
❌
❌
set
❌
❌
Functionality
CPU
GPU
Comments
bernoulli
❌
❌
multinomial
❌
❌
uniform
✅
✅
signed-uniform
✅
✅
rand normal
✅
✅
rand binary
✅
✅
Functionality
CPU
GPU
Comments
save
✅
✅
bin, onnx, png, jpg
load
✅
✅
bin, onnx, png, jpg
Functionality
CPU
GPU
Comments
abs
✅
✅
acos
✅
✅
add
✅
✅
asin
✅
✅
atan
✅
✅
ceil
✅
✅
clamp
✅
✅
clampmax
✅
✅
clampmin
✅
✅
cos
✅
✅
cosh
✅
✅
div
✅
✅
exp
✅
✅
floor
✅
✅
log
✅
✅
log2
✅
✅
log10
✅
✅
logn
✅
✅
max*
✅
❌
Not reduced
mean*
❌
❌
Not reduced
median*
❌
❌
Not reduced
min*
✅
❌
Not reduced
mod
✅
✅
mode*
✅
❌
Not reduced
mult
✅
✅
neg
✅
✅
normalize*
✅
✅
Not reduced
pow
✅
✅
reciprocal
✅
✅
remainder
✅
✅
round
✅
✅
rsqrt
✅
✅
sigmoid
✅
✅
sign
✅
✅
sin
✅
✅
sinh
✅
✅
sqr
✅
✅
sqrt
✅
✅
std*
❌
❌
Not reduced
sub
✅
✅
sum*
✅
✅
Not reduced by default
tan
✅
✅
tanh
✅
✅
trunc
✅
✅
var*
❌
❌
Not reduced
Functionality
CPU
GPU
Comments
argmax
❌
❌
argmin
❌
❌
cumprod
❌
❌
cumsum
❌
❌
max
❌
❌
min
❌
❌
mean
❌
❌
median
❌
❌
mode
❌
❌
norm
❌
❌
prod
❌
❌
std
❌
❌
sum
❌
❌
unique
❌
❌
var
❌
❌
Functionality
CPU
GPU
Comments
allclose
❌
❌
argsort
❌
❌
eq
✅
❌
ge
❌
❌
gt
❌
❌
isfinite
❌
❌
isinf
❌
❌
isnan
❌
❌
le
❌
❌
lt
❌
❌
ne
❌
❌
sort
❌
❌
topk
❌
❌
Functionality
CPU
GPU
Comments
cross
❌
❌
diag
❌
❌
einsum
❌
❌
flatten
❌
❌
flip
❌
❌
trace
❌
❌
dot
❌
❌