hodu 0.2.4

A user-friendly ML framework built in Rust for rapid prototyping and embedded deployment
Documentation
# ! <- no-backprop op
# ? <- combination of other ops
# @ <- alias
# {} <- group
# ^ <- non public

// binary
add
sub
mul
div
pow
maximum
minimum
// binary logical
logical_and!
logical_or!
logical_xor!
// cmp
eq!
ne!
lt!
le!
gt!
ge!
// cmp with scalar
eq_scalar!
ne_scalar!
lt_scalar!
le_scalar!
gt_scalar!
ge_scalar!
// unary - basic
neg
abs!
sign!
square
sqrt
recip
// unary - activation
relu
sigmoid
tanh
gelu
softplus
silu
swish@
mish
// unary - trigonometric
sin
cos
tan
// unary - exp
exp
exp2
exp10
ln
log2
log10
// unary logical
logical_not!
// unary with scalar - arithmetic
add_scalar
sub_scalar
mul_scalar
div_scalar
pow_scalar
maximum_scalar!
minimum_scalar!
// unary with scalar - activation
leaky_relu
elu
prelu
// matrix
matmul
dot
// reduce
sum
sum_all@
sum_to_shape@
mean
mean_all@
max!
min!
prod
std
std_all@
var
var_all@
norm?
l2_norm
l1_norm?
argmax!
argmin!
any!
all!
// concat
concat
cat@
stack?
// split
split
chunk?
// normalization
softmax?
log_softmax?
// indexing
index_select
index_put
gather
scatter
scatter_add
scatter_max
scatter_min
// selection
where3?
masked_fill?
clamp?
clamp_min?
clamp_max?
// conv
conv1d
conv2d
conv3d
conv_transpose1d
conv_transpose2d
conv_transpose3d
conv1d_grad_weight^
conv2d_grad_weight^
conv3d_grad_weight^
conv_transpose1d_grad_weight^
conv_transpose2d_grad_weight^
conv_transpose3d_grad_weight^
// windowing
{reduce_window} max!, mean, sum, min!
// shape
reshape
view@
flatten
squeeze
unsqueeze
broadcast
broadcast_like@
broadcast_left@
transpose
t@
permute
// shape with scalars
slice
// cast
to_dtype!
// memory
contiguous!