List of all items
Structs
- Variable
- VariableInner
- VariableWeak
- functions::transpose::TransposeByIndex
- nn::batch_norm::BatchNorm2dAutoGradConfig
- nn::batch_norm::BatchNorm2dInner
- nn::conv2d::Conv2dConfigs
- nn::conv2d::Conv2dConfigsInner
- nn::dropout::DropoutConfig
- nn::pool2d::MaxPool2dConfig
- nn::rnns::gru::GRUOutput
- nn::rnns::rnn::RNNOutput
- nn::rnns::weights::GRUCell
- nn::rnns::weights::LSTMCell
- nn::rnns::weights::RNNCell
- nn::rnns::weights::RNNLayerWeights
- nn::rnns::weights::RNNWeights
Traits
Functions
- activation::relu::relu
- activation::sigmoid::sigmoid
- concat::concat
- creator::alloc::alloc
- creator::alloc::alloc_like
- creator::from_vec::from_vec
- creator::ones::ones
- creator::ones::ones_like
- creator::rand::normal
- creator::rand::normal_like
- creator::rand::uniform
- creator::rand::uniform_like
- creator::zeros::zeros
- creator::zeros::zeros_like
- functions::broadcast::broadcast
- functions::clip::clip
- functions::cosh::cosh
- functions::exp::exp
- functions::flatten::flatten
- functions::index_axis::index_axis
- functions::log::log
- functions::matmul::matmul
- functions::powf::powf
- functions::reshape::reshape
- functions::sinh::sinh
- functions::slice::slice
- functions::stack::stack
- functions::sum::sum
- functions::sum_to::sum_to
- functions::tanh::tanh
- functions::transpose::transpose
- functions::transpose::transpose_by_index
- is_train
- loss::cross_entropy::cross_entropy
- loss::mse::mean_squared_error
- loss::softmax::softmax
- nn::batch_norm::batch_norm_2d
- nn::conv2d::conv2d
- nn::conv2d::conv2d_filter_grad
- nn::conv2d::deconv2d
- nn::dropout::dropout
- nn::pool2d::max_pool_2d
- nn::pool2d::max_pool_2d_grad
- nn::rnns::gru::naive::gru_naive
- nn::rnns::lstm::naive::lstm_naive
- nn::rnns::rnn::naive::rnn_relu
- nn::rnns::rnn::naive::rnn_tanh
- no_train
- set_train