Template Class ReflectionPadImpl#
Defined in File padding.h
Page Contents
Inheritance Relationships#
Base Type#
public torch::nn::Cloneable< Derived >(Template Class Cloneable)
Class Documentation#
-
template<size_t D, typename Derived>
class ReflectionPadImpl : public torch::nn::Cloneable<Derived># Base class for all (dimension-specialized) ReflectionPad modules.
Public Functions
-
inline ReflectionPadImpl(ExpandingArray<D * 2> padding)#
-
explicit ReflectionPadImpl(const ReflectionPadOptions<D> &options_)#
-
virtual void reset() override#
reset()must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.
-
Tensor forward(const Tensor &input)#
-
virtual void pretty_print(std::ostream &stream) const override#
Pretty prints the
ReflectionPad{1,2}dmodule into the givenstream.
Public Members
-
ReflectionPadOptions<D> options#
The options with which this
Modulewas constructed.
-
inline ReflectionPadImpl(ExpandingArray<D * 2> padding)#