Closed
Description
aten::prelu.Tensor
Appearance in TorchScript:
class PReLU(Module):
parameters = ["weight", ]
weight : Tensor
training : bool
def forward(self: torch.torch.nn.modules.activation.PReLU,
argument_1: Tensor) -> Tensor:
input = torch.prelu(argument_1, self.weight)
return input
Original PyTorch API: implicit PyTorch Tensor op
Relevant TensorRT Documentation: