|
|
@@ -347,6 +347,10 @@ class Print(PrimitiveWithInfer): |
|
|
... self.print('Print Tensor x and Tensor y:', x, y) |
|
|
... self.print('Print Tensor x and Tensor y:', x, y) |
|
|
... return x |
|
|
... return x |
|
|
... |
|
|
... |
|
|
|
|
|
>>> x = Tensor(np.ones([2, 1]).astype(np.int32)) |
|
|
|
|
|
>>> y = Tensor(np.ones([2, 2]).astype(np.int32)) |
|
|
|
|
|
>>> net = PrintDemo() |
|
|
|
|
|
>>> output = net(x, y) |
|
|
""" |
|
|
""" |
|
|
|
|
|
|
|
|
@prim_attr_register |
|
|
@prim_attr_register |
|
|
|