提交 69188e59 编写于 作者: T typhoonzero

fix ut

上级 c6d7c2bd
...@@ -19,6 +19,7 @@ from op_test import OpTest ...@@ -19,6 +19,7 @@ from op_test import OpTest
class TestSplitOp(OpTest): class TestSplitOp(OpTest):
def setUp(self): def setUp(self):
self._set_op_type()
axis = 1 axis = 1
x = np.random.random((4, 5, 6)).astype('float32') x = np.random.random((4, 5, 6)).astype('float32')
out = np.split(x, [2, 3], axis) out = np.split(x, [2, 3], axis)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册