Skip to main content

Module ops

Module ops 

Source
Expand description

Operations on tensors module.

Structs§

AttentionModuleOptions
Options for the attention module.
AvgPoolOptions
Average pooling options.
ConvOptions
Convolution options.
ConvTransposeOptions
Transposed convolution options.
DeformConvOptions
Deformable convolution options.
GridSampleOptions
Options for grid sampling operations.
InterpolateOptions
Interpolation options.
MaxPoolOptions
Max pooling options.
PaddedConvOptionsDeprecated
Convolution options with optional asymmetric end padding.
UnfoldOptions
Unfold operation options.

Enums§

GridSamplePaddingMode
Padding mode for grid sampling when coordinates are out of bounds.
IndexingUpdateOp
Computation to be used to update the existing values in indexed assignment operations (scatter/select).
InterpolateMode
Algorithm used.
PadMode
Padding mode for tensor pad operations.