Class LocalResponseNormImpl¶
Defined in File normalization.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::Cloneable< LocalResponseNormImpl >
(Template Class Cloneable)
Class Documentation¶
-
class LocalResponseNormImpl : public torch::nn::Cloneable<LocalResponseNormImpl>¶
Applies local response normalization over an input signal composed of several input planes, where channels occupy the second dimension.
Applies normalization across channels. See https://pytorch.org/docs/main/nn.html#torch.nn.LocalResponseNorm to learn about the exact behavior of this module.
See the documentation for
torch::nn::LocalResponseNormOptions
class to learn what constructor arguments are supported for this module.Example:
LocalResponseNorm model(LocalResponseNormOptions(2).alpha(0.0002).beta(0.85).k(2.));
Public Functions
-
inline LocalResponseNormImpl(int64_t size)¶
-
explicit LocalResponseNormImpl(const LocalResponseNormOptions &options_)¶
-
Tensor forward(const Tensor &input)¶
-
virtual void reset() override¶
reset()
must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.
-
virtual void pretty_print(std::ostream &stream) const override¶
Pretty prints the
LocalResponseNormImpl
module into the givenstream
.
Public Members
-
LocalResponseNormOptions options¶
The options with which this
Module
was constructed.
-
inline LocalResponseNormImpl(int64_t size)¶