Template Class InstanceNormImpl

• Defined in File instancenorm.h

Inheritance Relationships

Base Type

• public torch::nn::NormImplBase< D, Derived, InstanceNormOptions > (Template Class NormImplBase)

Class Documentation

template<size_t D, typename Derived>
class InstanceNormImpl : public torch::nn::NormImplBase<D, Derived, InstanceNormOptions>

Base class for all (dimension-specialized) instance norm modules.

Public Functions

inline Tensor forward(const Tensor &input)

inline virtual void pretty_print(std::ostream &stream) const override

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