Class Conv3dImpl

• Defined in File conv.h

Inheritance Relationships

Base Type

• public torch::nn::ConvNdImpl< 3, Conv3dImpl > (Template Class ConvNdImpl)

Class Documentation

class Conv3dImpl : public torch::nn::ConvNdImpl<3, Conv3dImpl>

Applies convolution over a 3-D input.

See https://pytorch.org/docs/main/nn.html#torch.nn.Conv3d to learn about the exact behavior of this module.

See the documentation for torch::nn::Conv3dOptions class to learn what constructor arguments are supported for this module.

Example:

Conv3d model(Conv3dOptions(3, 2, 3).stride(1).bias(false));

Public Functions

inline Conv3dImpl(int64_t input_channels, int64_t output_channels, ExpandingArray<3> kernel_size)

explicit Conv3dImpl(Conv3dOptions options_)

Tensor forward(const Tensor &input)