Class PReLUImpl#
Defined in File activation.h
Page Contents
Inheritance Relationships#
Base Type#
public torch::nn::Cloneable< PReLUImpl >
(Template Class Cloneable)
Class Documentation#
-
class PReLUImpl : public torch::nn::Cloneable<PReLUImpl>#
Applies the PReLU function element-wise.
See https://pytorch.org/docs/main/nn.html#torch.nn.PReLU to learn about the exact behavior of this module.
See the documentation for
torch::nn::PReLUOptions
class to learn what constructor arguments are supported for this module.Example:
PReLU model(PReLUOptions().num_parameters(42));
Public Functions
-
explicit PReLUImpl(const PReLUOptions &options_ = {})#
-
Tensor forward(const Tensor &input)#
Public Members
-
PReLUOptions options#
The options with which this
Module
was constructed.
-
Tensor weight#
The learned weight.
-
explicit PReLUImpl(const PReLUOptions &options_ = {})#