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: