Module primitiv::functions::node_funcs

Modules

batch
random

Functions

add
add_const
add_node
broadcast
concat
constant
constant_into
constant_on
conv2d
copy
copy_on
cos
divide
divide_const
divide_node
dropout
elu
exp
flatten
identity
identity_into
identity_on
input
input_into
input_on
log
log_softmax
logsumexp
lrelu
matmul
max_pool2d
mean
mean_nodes
multiply
multiply_const
multiply_node
negative
ones
ones_into
ones_on
parameter
parameter_into
pick
positive
pow
pow_const
pow_node
pown
prelu
relu
reshape
selu
sigmoid
sin
slice
softmax
softmax_cross_entropy
softmax_cross_entropy_with_ids
softplus
split
sqrt
stop_gradient
subtract
subtract_const
subtract_node
sum
sum_nodes
tan
tanh
transpose
zeros
zeros_into
zeros_on