pub trait ModuleOps<B>where
B: Backend,{
Show 31 methods
// Required methods
fn conv2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvOptions<2>,
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn conv3d(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvOptions<3>,
) -> <B as Backend>::FloatTensorPrimitive<5>;
fn conv_transpose2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvTransposeOptions<2>,
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn conv_transpose3d(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvTransposeOptions<3>,
) -> <B as Backend>::FloatTensorPrimitive<5>;
fn avg_pool2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn avg_pool2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
grad: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn adaptive_avg_pool2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
output_size: [usize; 2],
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn adaptive_avg_pool2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
grad: <B as Backend>::FloatTensorPrimitive<4>,
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn max_pool2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
dilation: [usize; 2],
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn max_pool2d_with_indices(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
dilation: [usize; 2],
) -> MaxPool2dWithIndices<B>;
fn max_pool2d_with_indices_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
dilation: [usize; 2],
output_grad: <B as Backend>::FloatTensorPrimitive<4>,
indices: <B as Backend>::IntTensorPrimitive<4>,
) -> MaxPool2dBackward<B>;
fn interpolate(
x: <B as Backend>::FloatTensorPrimitive<4>,
output_size: [usize; 2],
options: InterpolateOptions,
) -> <B as Backend>::FloatTensorPrimitive<4>;
fn interpolate_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
grad: <B as Backend>::FloatTensorPrimitive<4>,
output_size: [usize; 2],
options: InterpolateOptions,
) -> <B as Backend>::FloatTensorPrimitive<4>;
// Provided methods
fn embedding(
weights: <B as Backend>::FloatTensorPrimitive<2>,
indices: <B as Backend>::IntTensorPrimitive<2>,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn embedding_backward(
weights: <B as Backend>::FloatTensorPrimitive<2>,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
indices: <B as Backend>::IntTensorPrimitive<2>,
) -> <B as Backend>::FloatTensorPrimitive<2> { ... }
fn conv1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvOptions<1>,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn conv1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
options: ConvOptions<1>,
) -> Conv1dBackward<B> { ... }
fn conv2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<4>,
options: ConvOptions<2>,
) -> Conv2dBackward<B> { ... }
fn conv3d_backward(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<5>,
options: ConvOptions<3>,
) -> Conv3dBackward<B> { ... }
fn conv_transpose1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvTransposeOptions<1>,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn conv_transpose1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
options: ConvTransposeOptions<1>,
) -> Conv1dBackward<B> { ... }
fn conv_transpose2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<4>,
options: ConvTransposeOptions<2>,
) -> Conv2dBackward<B> { ... }
fn conv_transpose3d_backward(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<5>,
options: ConvTransposeOptions<3>,
) -> Conv3dBackward<B> { ... }
fn unfold4d(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
options: UnfoldOptions,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn avg_pool1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn avg_pool1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
grad: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn adaptive_avg_pool1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
output_size: usize,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn adaptive_avg_pool1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
grad: <B as Backend>::FloatTensorPrimitive<3>,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn max_pool1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
dilation: usize,
) -> <B as Backend>::FloatTensorPrimitive<3> { ... }
fn max_pool1d_with_indices(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
dilation: usize,
) -> MaxPool1dWithIndices<B> { ... }
fn max_pool1d_with_indices_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
dilation: usize,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
indices: <B as Backend>::IntTensorPrimitive<3>,
) -> MaxPool1dBackward<B> { ... }
}
Expand description
Module operations trait.
Required Methods§
fn conv2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvOptions<2>,
) -> <B as Backend>::FloatTensorPrimitive<4>
fn conv2d( x: <B as Backend>::FloatTensorPrimitive<4>, weight: <B as Backend>::FloatTensorPrimitive<4>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, options: ConvOptions<2>, ) -> <B as Backend>::FloatTensorPrimitive<4>
Two dimensional convolution.
§Shapes
x: [batch_size, channels_in, height, width]
,
weight: [channels_out, channels_in, kernel_size_1, kernel_size_2]
,
bias: [channels_out]
,
fn conv3d(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvOptions<3>,
) -> <B as Backend>::FloatTensorPrimitive<5>
fn conv3d( x: <B as Backend>::FloatTensorPrimitive<5>, weight: <B as Backend>::FloatTensorPrimitive<5>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, options: ConvOptions<3>, ) -> <B as Backend>::FloatTensorPrimitive<5>
Three dimensional convolution.
§Shapes
x: [batch_size, channels_in, depth, height, width]
,
weight: [channels_out, channels_in, kernel_size_1, kernel_size_2, kernel_size_3]
,
bias: [channels_out]
,
fn conv_transpose2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvTransposeOptions<2>,
) -> <B as Backend>::FloatTensorPrimitive<4>
fn conv_transpose2d( x: <B as Backend>::FloatTensorPrimitive<4>, weight: <B as Backend>::FloatTensorPrimitive<4>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, options: ConvTransposeOptions<2>, ) -> <B as Backend>::FloatTensorPrimitive<4>
Two dimensional transposed convolution.
§Shapes
x: [batch_size, channels_in, height, width]
,
weight: [channels_in, channels_out, kernel_size_1, kernel_size_2]
,
bias: [channels_out]
,
fn conv_transpose3d(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvTransposeOptions<3>,
) -> <B as Backend>::FloatTensorPrimitive<5>
fn conv_transpose3d( x: <B as Backend>::FloatTensorPrimitive<5>, weight: <B as Backend>::FloatTensorPrimitive<5>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, options: ConvTransposeOptions<3>, ) -> <B as Backend>::FloatTensorPrimitive<5>
Three dimensional transposed convolution.
§Shapes
x: [batch_size, channels_in, height, width]
,
weight: [channels_in, channels_out, kernel_size_1, kernel_size_2, kernel_size_3]
,
bias: [channels_out]
,
fn avg_pool2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<4>
fn avg_pool2d( x: <B as Backend>::FloatTensorPrimitive<4>, kernel_size: [usize; 2], stride: [usize; 2], padding: [usize; 2], count_include_pad: bool, ) -> <B as Backend>::FloatTensorPrimitive<4>
fn avg_pool2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
grad: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<4>
fn avg_pool2d_backward( x: <B as Backend>::FloatTensorPrimitive<4>, grad: <B as Backend>::FloatTensorPrimitive<4>, kernel_size: [usize; 2], stride: [usize; 2], padding: [usize; 2], count_include_pad: bool, ) -> <B as Backend>::FloatTensorPrimitive<4>
Backward pass for the avg pooling 2d operation.
fn adaptive_avg_pool2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
output_size: [usize; 2],
) -> <B as Backend>::FloatTensorPrimitive<4>
fn adaptive_avg_pool2d( x: <B as Backend>::FloatTensorPrimitive<4>, output_size: [usize; 2], ) -> <B as Backend>::FloatTensorPrimitive<4>
fn adaptive_avg_pool2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
grad: <B as Backend>::FloatTensorPrimitive<4>,
) -> <B as Backend>::FloatTensorPrimitive<4>
fn adaptive_avg_pool2d_backward( x: <B as Backend>::FloatTensorPrimitive<4>, grad: <B as Backend>::FloatTensorPrimitive<4>, ) -> <B as Backend>::FloatTensorPrimitive<4>
Backward pass for the adaptive avg pooling 2d operation.
fn max_pool2d(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
dilation: [usize; 2],
) -> <B as Backend>::FloatTensorPrimitive<4>
fn max_pool2d( x: <B as Backend>::FloatTensorPrimitive<4>, kernel_size: [usize; 2], stride: [usize; 2], padding: [usize; 2], dilation: [usize; 2], ) -> <B as Backend>::FloatTensorPrimitive<4>
fn max_pool2d_with_indices(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
dilation: [usize; 2],
) -> MaxPool2dWithIndices<B>
fn max_pool2d_with_indices( x: <B as Backend>::FloatTensorPrimitive<4>, kernel_size: [usize; 2], stride: [usize; 2], padding: [usize; 2], dilation: [usize; 2], ) -> MaxPool2dWithIndices<B>
fn max_pool2d_with_indices_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
stride: [usize; 2],
padding: [usize; 2],
dilation: [usize; 2],
output_grad: <B as Backend>::FloatTensorPrimitive<4>,
indices: <B as Backend>::IntTensorPrimitive<4>,
) -> MaxPool2dBackward<B>
fn max_pool2d_with_indices_backward( x: <B as Backend>::FloatTensorPrimitive<4>, kernel_size: [usize; 2], stride: [usize; 2], padding: [usize; 2], dilation: [usize; 2], output_grad: <B as Backend>::FloatTensorPrimitive<4>, indices: <B as Backend>::IntTensorPrimitive<4>, ) -> MaxPool2dBackward<B>
Backward pass for the max pooling 2d operation.
fn interpolate(
x: <B as Backend>::FloatTensorPrimitive<4>,
output_size: [usize; 2],
options: InterpolateOptions,
) -> <B as Backend>::FloatTensorPrimitive<4>
fn interpolate( x: <B as Backend>::FloatTensorPrimitive<4>, output_size: [usize; 2], options: InterpolateOptions, ) -> <B as Backend>::FloatTensorPrimitive<4>
fn interpolate_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
grad: <B as Backend>::FloatTensorPrimitive<4>,
output_size: [usize; 2],
options: InterpolateOptions,
) -> <B as Backend>::FloatTensorPrimitive<4>
fn interpolate_backward( x: <B as Backend>::FloatTensorPrimitive<4>, grad: <B as Backend>::FloatTensorPrimitive<4>, output_size: [usize; 2], options: InterpolateOptions, ) -> <B as Backend>::FloatTensorPrimitive<4>
Backward pass for the interpolate operation.
Provided Methods§
fn embedding(
weights: <B as Backend>::FloatTensorPrimitive<2>,
indices: <B as Backend>::IntTensorPrimitive<2>,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn embedding( weights: <B as Backend>::FloatTensorPrimitive<2>, indices: <B as Backend>::IntTensorPrimitive<2>, ) -> <B as Backend>::FloatTensorPrimitive<3>
fn embedding_backward(
weights: <B as Backend>::FloatTensorPrimitive<2>,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
indices: <B as Backend>::IntTensorPrimitive<2>,
) -> <B as Backend>::FloatTensorPrimitive<2>
fn embedding_backward( weights: <B as Backend>::FloatTensorPrimitive<2>, output_grad: <B as Backend>::FloatTensorPrimitive<3>, indices: <B as Backend>::IntTensorPrimitive<2>, ) -> <B as Backend>::FloatTensorPrimitive<2>
fn conv1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvOptions<1>,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn conv1d( x: <B as Backend>::FloatTensorPrimitive<3>, weight: <B as Backend>::FloatTensorPrimitive<3>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, options: ConvOptions<1>, ) -> <B as Backend>::FloatTensorPrimitive<3>
One dimensional convolution.
§Shapes
x: [batch_size, channels_in, length]
,
weight: [channels_out, channels_in, kernel_size]
,
bias: [channels_out]
,
fn conv1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
options: ConvOptions<1>,
) -> Conv1dBackward<B>
fn conv1d_backward( x: <B as Backend>::FloatTensorPrimitive<3>, weight: <B as Backend>::FloatTensorPrimitive<3>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, output_grad: <B as Backend>::FloatTensorPrimitive<3>, options: ConvOptions<1>, ) -> Conv1dBackward<B>
Backward pass for the conv1d operation.
fn conv2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<4>,
options: ConvOptions<2>,
) -> Conv2dBackward<B>
fn conv2d_backward( x: <B as Backend>::FloatTensorPrimitive<4>, weight: <B as Backend>::FloatTensorPrimitive<4>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, output_grad: <B as Backend>::FloatTensorPrimitive<4>, options: ConvOptions<2>, ) -> Conv2dBackward<B>
Backward pass for the conv2d operation.
fn conv3d_backward(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<5>,
options: ConvOptions<3>,
) -> Conv3dBackward<B>
fn conv3d_backward( x: <B as Backend>::FloatTensorPrimitive<5>, weight: <B as Backend>::FloatTensorPrimitive<5>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, output_grad: <B as Backend>::FloatTensorPrimitive<5>, options: ConvOptions<3>, ) -> Conv3dBackward<B>
Backward pass for the conv3d operation.
fn conv_transpose1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
options: ConvTransposeOptions<1>,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn conv_transpose1d( x: <B as Backend>::FloatTensorPrimitive<3>, weight: <B as Backend>::FloatTensorPrimitive<3>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, options: ConvTransposeOptions<1>, ) -> <B as Backend>::FloatTensorPrimitive<3>
One dimensional transposed convolution.
§Shapes
x: [batch_size, channels_in, length]
,
weight: [channels_in, channels_out, length]
,
bias: [channels_out]
,
fn conv_transpose1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
weight: <B as Backend>::FloatTensorPrimitive<3>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
options: ConvTransposeOptions<1>,
) -> Conv1dBackward<B>
fn conv_transpose1d_backward( x: <B as Backend>::FloatTensorPrimitive<3>, weight: <B as Backend>::FloatTensorPrimitive<3>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, output_grad: <B as Backend>::FloatTensorPrimitive<3>, options: ConvTransposeOptions<1>, ) -> Conv1dBackward<B>
Backward pass for the conv transpose 1d operation.
fn conv_transpose2d_backward(
x: <B as Backend>::FloatTensorPrimitive<4>,
weight: <B as Backend>::FloatTensorPrimitive<4>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<4>,
options: ConvTransposeOptions<2>,
) -> Conv2dBackward<B>
fn conv_transpose2d_backward( x: <B as Backend>::FloatTensorPrimitive<4>, weight: <B as Backend>::FloatTensorPrimitive<4>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, output_grad: <B as Backend>::FloatTensorPrimitive<4>, options: ConvTransposeOptions<2>, ) -> Conv2dBackward<B>
Backward pass for the conv transpose 2d operation.
fn conv_transpose3d_backward(
x: <B as Backend>::FloatTensorPrimitive<5>,
weight: <B as Backend>::FloatTensorPrimitive<5>,
bias: Option<<B as Backend>::FloatTensorPrimitive<1>>,
output_grad: <B as Backend>::FloatTensorPrimitive<5>,
options: ConvTransposeOptions<3>,
) -> Conv3dBackward<B>
fn conv_transpose3d_backward( x: <B as Backend>::FloatTensorPrimitive<5>, weight: <B as Backend>::FloatTensorPrimitive<5>, bias: Option<<B as Backend>::FloatTensorPrimitive<1>>, output_grad: <B as Backend>::FloatTensorPrimitive<5>, options: ConvTransposeOptions<3>, ) -> Conv3dBackward<B>
Backward pass for the conv transpose 3d operation.
fn unfold4d(
x: <B as Backend>::FloatTensorPrimitive<4>,
kernel_size: [usize; 2],
options: UnfoldOptions,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn unfold4d( x: <B as Backend>::FloatTensorPrimitive<4>, kernel_size: [usize; 2], options: UnfoldOptions, ) -> <B as Backend>::FloatTensorPrimitive<3>
Four-dimensional unfolding.
§Shapes
x: [batch_size, channels_in, height, width]
,
returns: [batch_size, channels_in * kernel_size_1 * kernel_size_2, number of blocks]
,
fn avg_pool1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn avg_pool1d( x: <B as Backend>::FloatTensorPrimitive<3>, kernel_size: usize, stride: usize, padding: usize, count_include_pad: bool, ) -> <B as Backend>::FloatTensorPrimitive<3>
fn avg_pool1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
grad: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
count_include_pad: bool,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn avg_pool1d_backward( x: <B as Backend>::FloatTensorPrimitive<3>, grad: <B as Backend>::FloatTensorPrimitive<3>, kernel_size: usize, stride: usize, padding: usize, count_include_pad: bool, ) -> <B as Backend>::FloatTensorPrimitive<3>
Backward pass for the avg pooling 1d operation.
fn adaptive_avg_pool1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
output_size: usize,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn adaptive_avg_pool1d( x: <B as Backend>::FloatTensorPrimitive<3>, output_size: usize, ) -> <B as Backend>::FloatTensorPrimitive<3>
fn adaptive_avg_pool1d_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
grad: <B as Backend>::FloatTensorPrimitive<3>,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn adaptive_avg_pool1d_backward( x: <B as Backend>::FloatTensorPrimitive<3>, grad: <B as Backend>::FloatTensorPrimitive<3>, ) -> <B as Backend>::FloatTensorPrimitive<3>
Backward pass for the adaptive avg pooling 1d operation.
fn max_pool1d(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
dilation: usize,
) -> <B as Backend>::FloatTensorPrimitive<3>
fn max_pool1d( x: <B as Backend>::FloatTensorPrimitive<3>, kernel_size: usize, stride: usize, padding: usize, dilation: usize, ) -> <B as Backend>::FloatTensorPrimitive<3>
fn max_pool1d_with_indices(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
dilation: usize,
) -> MaxPool1dWithIndices<B>
fn max_pool1d_with_indices( x: <B as Backend>::FloatTensorPrimitive<3>, kernel_size: usize, stride: usize, padding: usize, dilation: usize, ) -> MaxPool1dWithIndices<B>
fn max_pool1d_with_indices_backward(
x: <B as Backend>::FloatTensorPrimitive<3>,
kernel_size: usize,
stride: usize,
padding: usize,
dilation: usize,
output_grad: <B as Backend>::FloatTensorPrimitive<3>,
indices: <B as Backend>::IntTensorPrimitive<3>,
) -> MaxPool1dBackward<B>
fn max_pool1d_with_indices_backward( x: <B as Backend>::FloatTensorPrimitive<3>, kernel_size: usize, stride: usize, padding: usize, dilation: usize, output_grad: <B as Backend>::FloatTensorPrimitive<3>, indices: <B as Backend>::IntTensorPrimitive<3>, ) -> MaxPool1dBackward<B>
Backward pass for the max pooling 1d operation.