summaryrefslogtreecommitdiff
path: root/res/PyTorchExamples/examples/lt/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'res/PyTorchExamples/examples/lt/__init__.py')
-rw-r--r--res/PyTorchExamples/examples/lt/__init__.py17
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)]