Template Class ConvNdImpl#
Defined in File conv.h
Page Contents
Inheritance Relationships#
Base Type#
public torch::nn::Cloneable< Derived >(Template Class Cloneable)
Derived Types#
public torch::nn::ConvTransposeNdImpl< 1, ConvTranspose1dImpl >(Template Class ConvTransposeNdImpl)public torch::nn::ConvTransposeNdImpl< 3, ConvTranspose3dImpl >(Template Class ConvTransposeNdImpl)public torch::nn::ConvTransposeNdImpl< 2, ConvTranspose2dImpl >(Template Class ConvTransposeNdImpl)public torch::nn::ConvTransposeNdImpl< D, Derived >(Template Class ConvTransposeNdImpl)
Class Documentation#
-
template<size_t D, typename Derived>
class ConvNdImpl : public torch::nn::Cloneable<Derived># Base class for all (dimension-specialized) convolution modules.
Subclassed by torch::nn::ConvTransposeNdImpl< 1, ConvTranspose1dImpl >, torch::nn::ConvTransposeNdImpl< 3, ConvTranspose3dImpl >, torch::nn::ConvTransposeNdImpl< 2, ConvTranspose2dImpl >, torch::nn::ConvTransposeNdImpl< D, Derived >
Public Functions
-
inline explicit ConvNdImpl(detail::ConvNdOptions<D> options_)#
-
inline virtual void reset() override#
reset()must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.
-
inline void reset_parameters()#
-
inline virtual void pretty_print(std::ostream &stream) const override#
Pretty prints the
Conv{1,2,3}dmodule into the givenstream.
Public Members
-
detail::ConvNdOptions<D> options#
The options with which this
Modulewas constructed.
-
Tensor weight#
The learned kernel (or “weight”).
-
Tensor bias#
The learned bias. Only defined if the
biasoption was true.
Protected Attributes
-
std::vector<int64_t> _reversed_padding_repeated_twice#
-
inline explicit ConvNdImpl(detail::ConvNdOptions<D> options_)#