Module juice::layers::common
[−]
[src]
Provides common neural network layers.
For now the layers in common should be discribed as layers that are typical layers for building neural networks but are not activation or loss layers.
Re-exports
pub use self::convolution::Convolution; |
pub use self::convolution::ConvolutionConfig; |
pub use self::linear::Linear; |
pub use self::linear::LinearConfig; |
pub use self::log_softmax::LogSoftmax; |
pub use self::pooling::Pooling; |
pub use self::pooling::PoolingConfig; |
pub use self::pooling::PoolingMode; |
pub use self::softmax::Softmax; |
Modules
convolution |
Convolves the input tensor. |
linear |
Applies a linear transformation to the input data |
log_softmax |
Computes the logarithmic softmax of its input. |
pooling |
Applies pooling to the input. |
softmax |
Computes the softmax of its input. |
Traits
FilterLayer |
Provides common utilities for Layers that utilize a filter with stride and padding. |