mindspore.ops.TensorSummary

class mindspore.ops.TensorSummary

将Tensor保存到Summary文件。必须和SummaryRecord或SummaryCollector一起使用， Summary文件的保存路径由SummaryRecord或SummaryCollector指定。Summary文件可以通过MindInsight加载并展示， 关于MindInsight的详细信息请参考 MindInsight文档 。

输入：

• name (str) - 输入变量的名称。

• value (Tensor) - Tensor的值，Tensor的维度必须大于0。

异常：

• TypeError - 如果 name 不是str。

• TypeError - 如果 value 不是Tensor。

• ValueError - 如果 value 的维度等于0。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import mindspore.nn as nn
>>> import mindspore.ops as ops
>>> from mindspore import Tensor, set_context
>>>
>>>
>>> class SummaryDemo(nn.Cell):
...     def __init__(self,):
...         super(SummaryDemo, self).__init__()
...         self.summary = ops.TensorSummary()
...         self.add = ops.Add()
...
...     def construct(self, x, y):
...         x = self.add(x, y)
...         name = "x"
...         self.summary(name, x)
...         return x
>>> set_context(mode=mindspore.GRAPH_MODE)
>>> summary = SummaryDemo()(Tensor([[1]]), Tensor([[2]]))
>>> print(summary)
[[3]]