Rate this Page

Template Class BatchNormImplBase#

Inheritance Relationships#

Base Type#

Class Documentation#

template<size_t D, typename Derived>
class BatchNormImplBase : public torch::nn::NormImplBase<D, Derived, BatchNormOptions>#

Base class for all (dimension-specialized) batchnorm modules.

Public Functions

inline Tensor forward(const Tensor &input)#
inline virtual void pretty_print(std::ostream &stream) const override#

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