Module burn::nn

source · []

Structs

Set at random some elements of the input tensor to zero during training.
Configuration to create a Dropout layer.
Applies the Gaussian Error Linear Units function element-wise.
Applies Layer Normalization over an input tensor as described in the paper Layer Normalization.
Configuration to create a LayerNorm layer.
Applies a linear transformation to the input tensor:
Configuration to create a Linear layer.
Applies the rectified linear unit function element-wise: