mindspore.ops.HistogramSummary
- class mindspore.ops.HistogramSummary[源代码]
计算Tensor的直方图并保存到Summary文件。必须和SummaryRecord或SummaryCollector一起使用, Summary文件的保存路径由SummaryRecord或SummaryCollector指定。Summary文件可以通过MindInsight加载并展示, 关于MindInsight的详细信息请参考 MindInsight文档 。
- 输入:
name (str) - 输入变量的名称。
value (Tensor) - Tensor的值,Tensor的rank必须大于0。
- 异常:
TypeError - 如果 name 不是str。
TypeError - 如果 value 不是Tensor。
- 支持平台:
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.HistogramSummary() ... 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, 2]), Tensor([3, 4])) >>> print(summary) [4 6]