Template Class ConvTransposeNdImpl¶
Defined in File conv.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::ConvNdImpl< D, Derived >(Template Class ConvNdImpl)
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}dmodule into the givenstream.
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)¶
-
inline explicit ConvTransposeNdImpl(detail::ConvNdOptions<D> options_)¶
