Skip to main content
Back to top
Ctrl
+
K
API Reference
Binary
add
div
equal
floordiv
greater_equal
maximum
minimum
mod
mul
not_equal
pow
sub
Creation
arange
array
full_like
glorot_uniform
he_normal
he_uniform
lecun_normal
lecun_uniform
ndarange
ndarange_like
ones
ones_like
rand
rand_like
randn
randn_like
triu
xavier_normal
xavier_uniform
zeros
zeros_like
Indexing
gather
scatter
Linalg
matmul
Reduce
argmax
max
mean
sum
sum_batch_dims
Special
cond
logsumexp
softmax
where
Unary
abs
cast
cos
decr_batch_dim_ctr
exp
floor
incr_batch_dim_ctr
log
logical_not
negate
relu
sigmoid
sin
sqrt
tanh
transfer_to
View
array_slice
broadcast_batch_dims
broadcast_to
concatenate
move_axis_from_front
move_axis_from_front_of_batch_dims
move_axis_to_front
move_axis_to_front_of_batch_dims
pad
permute
permute_batch_dims
reshape
shallow_copy
squeeze
squeeze_batch_dims
stack
transpose
unsqueeze
unsqueeze_batch_dims
Tutorials
A Primer on Automatic Differentiation
Training an MLP (CPU and GPU)
Training a Transformer (CPU and GPU)
Program Transformations Uncovered
Repository
Suggest edit
Open issue
.rst
.pdf
Linalg
Linalg
#
Functions
matmul