Skip to main content
Back to top
Ctrl
+
K
API Reference
Array Class
Array
Function Transformations
djit
grad
jacfwd
jacrev
jit
jvp
vjp
vmap
xpr
Array 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
Unary Operations
abs
cast
cos
decr_batch_dim_ctr
exp
floor
incr_batch_dim_ctr
log
logical_not
negate
relu
sigmoid
sin
sqrt
tanh
transfer_to
Binary Operations
add
div
equal
floordiv
greater_equal
maximum
minimum
mod
mul
not_equal
pow
sub
Reduction Operations
argmax
max
mean
sum
sum_batch_dims
Linear Algebra
matmul
Array Manipulation
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
Understanding Nabla - Program Transformations (Part 1)
Value-and-Grads (CPU)
Value-and-Grads (GPU)
MLP Training (CPU)
JAX vs. Nabla: Training an MLP (CPU)
JAX vs. Nabla: Training a Transformer (CPU)
Repository
Suggest edit
Open issue
.md
.pdf
equal
Contents
Signature
Description
equal
#
Signature
#
nabla
.
equal
(
arg0
,
arg1
)
->
'Array'
Description
#
Element-wise equality comparison.
Contents
Signature
Description