Module burn_core::tensor::ops

Operations on tensors module.

Modules

• conv
  Module with convolution operations.
• pool
  Module with pooling operations.

Structs

• Conv1dBackward
  Gradient computed during the backward pass for each tensor used by conv1d.
• Conv2dBackward
  Gradient computed during the backward pass for each tensor used by conv2d.
• ConvOptions
  Convolution options.
• ConvTransposeOptions
  Transposed convolution options.
• MaxPool1dBackward
  Gradient computed during the backward pass for each tensor used by max_pool1d.
• MaxPool1dWithIndices
  Results from max_pool1d.
• MaxPool2dBackward
  Gradient computed during the backward pass for each tensor used by max_pool2d.
• MaxPool2dWithIndices
  Results from max_pool2d.
• UnfoldOptions
  Unfold operation options.

Traits

• ActivationOps
  Activation function operations.
• BoolTensorOps
  Bool Tensor API for basic operations, see tensor for documentation on each function.
• IntTensorOps
  Int Tensor API for basic and numeric operations, see tensor for documentation on each function.
• ModuleOps
  Module operations trait.
• TensorOps
  Operations on float tensors.