update test

pytorch · Jan 28, 2021 · dd1d590 · dd1d590
1 parent d5a7c61
commit dd1d590
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/onnx/test_pytorch_onnx_onnxruntime.py b/test/onnx/test_pytorch_onnx_onnxruntime.py
@@ -2742,6 +2742,7 @@ def __init__(self):
                 self.register_buffer("weight", torch.ones(5))
 
             def forward(self, x):
+                # shape is of rank 0
                 shape = self.weight.shape[0]
                 m = 5 - shape
                 return x.clamp(min=m)
@@ -2755,6 +2756,7 @@ def __init__(self):
                 self.register_buffer("weight", torch.ones(2))
 
             def forward(self, x):
+                # shape is of rank 0
                 shape = self.weight.shape[0]
                 pad = [1, shape, shape, shape]
                 zero_pad = torch.nn.ZeroPad2d(pad)