diff options
Diffstat (limited to 'res/PyTorchExamples/examples/PReLU/__init__.py')
-rw-r--r-- | res/PyTorchExamples/examples/PReLU/__init__.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/res/PyTorchExamples/examples/PReLU/__init__.py b/res/PyTorchExamples/examples/PReLU/__init__.py new file mode 100644 index 000000000..6c0f7a2d7 --- /dev/null +++ b/res/PyTorchExamples/examples/PReLU/__init__.py @@ -0,0 +1,18 @@ +import torch +import torch.nn as nn + + +# model +class net_PReLU(nn.Module): + def __init__(self): + super().__init__() + self.op = nn.PReLU() + + def forward(self, input): + return self.op(input) + + +_model_ = net_PReLU() + +# dummy input for onnx generation +_dummy_ = torch.randn(1, 2, 3, 3) |