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