Rate this Page

Template Class ConvTransposeNdImpl#

Inheritance Relationships#

Base Type#

Class Documentation#

template<size_t D, typename Derived>
class ConvTransposeNdImpl : public torch::nn::ConvNdImpl<D, Derived>#

Base class for all (dimension-specialized) convolution transpose modules.

Public Functions

inline explicit ConvTransposeNdImpl(detail::ConvNdOptions<D> options_)#
inline virtual void pretty_print(std::ostream &stream) const override#

Pretty prints the ConvTranspose{1,2,3}d module into the given stream.

Protected Functions

inline const ExpandingArray<D> &padding() const#
std::vector<int64_t> _output_padding(const Tensor &input, const std::optional<at::IntArrayRef> &output_size, const ExpandingArray<D> &stride, const ExpandingArray<D> &padding, const ExpandingArray<D> &kernel_size)#