Module burn_tensor::ops

Modules

• conv

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.
• MaxPool2dBackward
  Gradient computed during the backward pass for each tensor used by max_pool2d.
• MaxPool2dWithIndexes
  Results from max_pool2d.

Traits

• 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
• TensorOps
  Operations on float tensors.