mindspore.ops.Print
- class mindspore.ops.Print[源代码]
将输入Tensor或string进行打印输出。
默认打印在屏幕上。也可以保存在文件中,通过 context 设置 print_file_path 参数。一旦设置,输出将保存在指定文件中。通过函数
mindspore.parse_print()
可以重新加载数据。获取更多信息,请查看mindspore.context.set_context()
和mindspore.parse_print()
。Note
在PyNative模式下,请使用Python print函数。在Graph模式下,bool、int和float将被转换为Tensor进行打印,str保持不变。 该方法用于代码调试,当同时print大量数据时,为了保证主进程不受影响,可能会丢失一些数据,这时推荐使用 Summary 功能,具体可查看 Summary.
输入:
input_x (Union[Tensor, bool, int, float, str]) - Print的输入。支持多个输入,用’,’分隔。
输出:
Tensor,数据类型和shape与 input_x 相同。
异常:
TypeError - input_x 不是Tensor、bool、int、float或str。
- 支持平台:
Ascend
GPU
样例:
>>> class PrintDemo(nn.Cell): ... def __init__(self): ... super(PrintDemo, self).__init__() ... self.print = ops.Print() ... ... def construct(self, x, y): ... self.print('Print Tensor x and Tensor y:', x, y) ... return x ... >>> x = Tensor(np.ones([2, 1]).astype(np.int32)) >>> y = Tensor(np.ones([2, 2]).astype(np.int32)) >>> net = PrintDemo() >>> result = net(x, y) Print Tensor x and Tensor y: Tensor(shape=[2, 1], dtype=Int32, value= [[1] [1]]) Tensor(shape=[2, 2], dtype=Int32, value= [[1 1] [1 1]])