Adding tests for Transpose op

docs/Operators.md

@@ -6298,6 +6298,52 @@ opset_import {
 </dl>
 
 
+#### Examples
+
+<details>
+<summary>all_permutations</summary>
+
+```python
+shape = (2,3,4)
+data = np.random.random_sample(shape).astype(np.float32)
+permutations = list(itertools.permutations(np.arange(len(shape))))
+
+for i in range(len(permutations)):
+    node = onnx.helper.make_node(
+        'Transpose',
+        inputs=['data'],
+        outputs=['transposed'],
+        perm=permutations[i]
+    )            
+    transposed = np.transpose(data, permutations[i])
+    expect(node, inputs=[data], outputs=[transposed],
+        name='test_transpose_all_permutations_' + str(i))            
+```
+
+</details>
+
+
+<details>
+<summary>default</summary>
+
+```python
+shape = (2, 3, 4)
+data = np.random.random_sample(shape).astype(np.float32)
+
+node = onnx.helper.make_node(
+    'Transpose',
+    inputs=['data'],
+    outputs=['transposed']
+)
+
+transposed = np.transpose(data)
+expect(node, inputs=[data], outputs=[transposed],
+    name='test_transpose_default')
+```
+
+</details>
+
+
 ### <a name="Unsqueeze"></a><a name="unsqueeze">**Unsqueeze**</a>
 
   Insert single-dimensional entries to the shape of a tensor.

onnx/backend/test/case/node/transpose.py

@@ -0,0 +1,46 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+import numpy as np
+import itertools
+
+import onnx
+from ..base import Base
+from . import expect
+
+
+class Transpose(Base):
+
+    @staticmethod
+    def export_default():
+        shape = (2, 3, 4)
+        data = np.random.random_sample(shape).astype(np.float32)
+
+        node = onnx.helper.make_node(
+            'Transpose',
+            inputs=['data'],
+            outputs=['transposed']
+        )
+
+        transposed = np.transpose(data)
+        expect(node, inputs=[data], outputs=[transposed],
+            name='test_transpose_default')
+
+    @staticmethod
+    def export_all_permutations():
+        shape = (2,3,4)
+        data = np.random.random_sample(shape).astype(np.float32)
+        permutations = list(itertools.permutations(np.arange(len(shape))))
+
+        for i in range(len(permutations)):
+            node = onnx.helper.make_node(
+                'Transpose',
+                inputs=['data'],
+                outputs=['transposed'],
+                perm=permutations[i]
+            )            
+            transposed = np.transpose(data, permutations[i])
+            expect(node, inputs=[data], outputs=[transposed],
+                name='test_transpose_all_permutations_' + str(i))            

onnx/backend/test/data/node/test_transpose_all_permutations_0/input_0.pb

@@ -0,0 +1 @@
+
BdataJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_0/output_0.pb

@@ -0,0 +1,2 @@
+
B
+transposedJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_1/input_0.pb

@@ -0,0 +1 @@
+
BdataJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_1/output_0.pb

@@ -0,0 +1,2 @@
+
B
+transposedJ`?H��>��v?�7?QY%?rR�>�N?n�>~�J?w}?<Kd?�e?^k?��<��z?��l?�&U?��L?Z{�=H5G?G�>�p�=�^?��G?

onnx/backend/test/data/node/test_transpose_all_permutations_2/input_0.pb

@@ -0,0 +1 @@
+
BdataJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_2/output_0.pb

@@ -0,0 +1,2 @@
+
B
+transposedJ`?�7?�N?w}?^k?��l?Z{�=�p�=H��>QY%?n�><Kd?��<�&U?H5G?�^?��v?rR�>~�J?�e?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_3/input_0.pb

@@ -0,0 +1 @@
+
BdataJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_3/output_0.pb

@@ -0,0 +1,2 @@
+
B
+transposedJ`?^k?�7?��l?�N?Z{�=w}?�p�=H��>��<QY%?�&U?n�>H5G?<Kd?�^?��v?��z?rR�>��L?~�J?G�>�e?��G?

onnx/backend/test/data/node/test_transpose_all_permutations_4/input_0.pb

@@ -0,0 +1 @@
+
BdataJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_4/output_0.pb

@@ -0,0 +1,2 @@
+
B
+transposedJ`?H��>��v?^k?��<��z?�7?QY%?rR�>��l?�&U?��L?�N?n�>~�J?Z{�=H5G?G�>w}?<Kd?�e?�p�=�^?��G?

onnx/backend/test/data/node/test_transpose_all_permutations_5/input_0.pb

@@ -0,0 +1 @@
+
BdataJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_all_permutations_5/output_0.pb

@@ -0,0 +1,2 @@
+
B
+transposedJ`?^k?H��>��<��v?��z?�7?��l?QY%?�&U?rR�>��L?�N?Z{�=n�>H5G?~�J?G�>w}?�p�=<Kd?�^?�e?��G?

onnx/backend/test/data/node/test_transpose_default/input_0.pb

@@ -0,0 +1 @@
+
BdataJ`?�7?�N?w}?H��>QY%?n�><Kd?��v?rR�>~�J?�e?^k?��l?Z{�=�p�=��<�&U?H5G?�^?��z?��L?G�>��G?

onnx/backend/test/data/node/test_transpose_default/node.pb

@@ -0,0 +1,3 @@
+
+data
+transposed"	Transpose

onnx/backend/test/data/node/test_transpose_default/output_0.pb

@@ -0,0 +1,2 @@
+
B
+transposedJ`?^k?H��>��<��v?��z?�7?��l?QY%?�&U?rR�>��L?�N?Z{�=n�>H5G?~�J?G�>w}?�p�=<Kd?�^?�e?��G?