Module burn_tensor::module

Functions

• conv1d
  Applies a 1D convolution.
• conv2d
  Applies a 2D convolution.
• embedding
  Applies the embedding module.
• max_pool2d
  Applies a 2D max pooling.
• max_pool2d_with_indexes
  Applies a 2D max pooling with indexes.