def export_inverse_batched(): # type: () -> None node = onnx.helper.make_node('Inverse', inputs=['x'], outputs=['y']) X = np.random.randn(2, 3, 4, 4) Y = inverse_reference_implementation(X) expect(node, inputs=[X], outputs=[Y], name='test_inverse_batched')
def export_squeeze(): # type: () -> None node = onnx.helper.make_node( 'Squeeze', inputs=['x', 'axes'], outputs=['y'], ) x = np.random.randn(1, 3, 4, 5).astype(np.float32) axes = np.array([0], dtype=np.int64) y = np.squeeze(x, axis=0) expect(node, inputs=[x, axes], outputs=[y], name='test_squeeze')
def export_squeeze_negative_axes() -> None: node = onnx.helper.make_node( 'Squeeze', inputs=['x', 'axes'], outputs=['y'], ) x = np.random.randn(1, 3, 1, 5).astype(np.float32) axes = np.array([-2], dtype=np.int64) y = np.squeeze(x, axis=-2) expect(node, inputs=[x, axes], outputs=[y], name='test_squeeze_negative_axes')
def export_squeeze_negative_axes(): # type: () -> None node = onnx.helper.make_node( 'Squeeze', inputs=['x'], outputs=['y'], axes=[-2], ) x = np.random.randn(1, 3, 1, 5).astype(np.float32) y = np.squeeze(x, axis=-2) expect(node, inputs=[x], outputs=[y], name='test_squeeze_negative_axes')