Module primitiv::functions::tensor_funcs [−] [src]

Modules

batch
random

Functions

add
add_const
add_tensor
broadcast
concat
constant
constant_on
conv2d
copy
copy_on
cos
divide
divide_const
divide_tensor
dropout
elu
exp
flatten
identity
identity_on
input
input_on
log
log_softmax
logsumexp
lrelu
matmul
max_pool2d
mean
mean_tensors
multiply
multiply_const
multiply_tensor
negative
ones
ones_on
parameter
pick
positive
pow
pow_const
pow_tensor
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_tensor
sum
sum_tensors
tan
tanh
transpose
zeros
zeros_on