mindspore.ops.TensorDump

class mindspore.ops.TensorDump(input_output='out')

将Tensor保存为numpy格式的npy文件。

警告

• 参数input_output将不再支持参数值为'all'。

说明

• 在Ascend平台上的Graph模式下，可以通过设置环境变量 MS_DUMP_SLICE_SIZE 和 MS_DUMP_WAIT_TIME 解决在输出大Tesnor或输出Tensor比较密集场景下算子执行失败的问题。

参数：

• input_output (str，可选) - 控制Tensordump行为模式的参数，可选的值为 ['out', 'in'] 中的一个，默认值： out 。

  对于算子A –> 重排算子 –> 算子B这样的情况，由于重排算子的插入，导致算子A的输出不再等价于算子B的输入。

  假设一种情况是算子A的计算结果既作为算子B的输入，也作为Tensordump算子的输入。则在该情况下， 通过设置input_output参数可以实现不同的保存数据的需求：

  • 如果input_output参数设置为'out'，保存的数据仅包含算子A的输出分片。

  • 如果input_output参数设置为'in'，保存的数据将仅包含算子B的输入分片。

  当input_output参数被配置为'in'时，生成的输入分片所对应的npy文件命名格式为：fileName_dumpMode_dtype_id.npy。

  当input_output参数被配置为'out'时，生成的输出分片所对应的npy文件命名格式为：fileName_dtype_id.npy。

  • fileName：参数file的值 （若该参数传入时是一个使用者指定的路径，则fileName的值为路径的最后一级）。

  • dumpMode：input_output参数的值。

  • dtype：原始的数据类型。bfloat16类型数据保存在.npy文件中会被转换成float32类型。

  • id：一个自增的ID。

输入：

• file (str) - 要保存的文件路径。

• input_x (Tensor) - 任意维度的Tensor。

异常：

• TypeError - 如果 file 不是str。

• TypeError - 如果 input_x 不是Tensor。

支持平台：

Ascend

样例：

>>> import numpy as np
>>> import mindspore as ms
>>> import time
>>> from mindspore import nn, Tensor, ops
>>> ms.set_context(mode=ms.GRAPH_MODE, device_target="Ascend")
>>> class Net(nn.Cell):
...     def __init__(self):
...         super(Net, self).__init__()
...         self.dump = ops.TensorDump()
...
...     def construct(self, x):
...         x += 1.
...         self.dump('add', x)
...         x /= 2.
...         self.dump('div', x)
...         x *= 5.
...         self.dump('mul', x)
...         return x
...
>>> x = np.array([[1, 2, 3, 4], [5, 6, 7, 8]]).astype(np.float32)
>>> input_x = Tensor(x)
>>> net = Net()
>>> out = net(input_x)
>>> time.sleep(0.5)
>>> add = np.load('add_float32_0.npy')
>>> print(add)
[[2. 3. 4. 5.]
 [6. 7. 8. 9.]]