mindspore.ops.ImageSummary
- class mindspore.ops.ImageSummary[源代码]
将图像保存到Summary文件。必须和SummaryRecord或SummaryCollector一起使用, Summary文件的保存路径由SummaryRecord或SummaryCollector指定。Summary文件可以通过MindInsight加载并展示, 关于MindInsight的详细信息请参考 MindInsight文档 。
- 输入:
name (str) - 输入变量的名称,不能是空字符串。
value (Tensor) - 图像数据的值,Tensor的rank必须为4。
- 异常:
TypeError - 如果 name 不是str。
TypeError - 如果 value 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.nn as nn >>> import mindspore.ops as ops >>> >>> >>> class Net(nn.Cell): ... def __init__(self): ... super(Net, self).__init__() ... self.summary = ops.ImageSummary() ... ... def construct(self, x): ... name = "image" ... out = self.summary(name, x) ... return out ...