mindspore.ops.TensorSummary

class mindspore.ops.TensorSummary(*args, **kwargs)

Outputs a tensor to a protocol buffer through a tensor summary operator.

Inputs:

• name (str) - The name of the input variable.

• value (Tensor) - The value of tensor, and the rank of tensor must be greater than 0.

Raises

• TypeError – If name is not a str.

• TypeError – If value is not a Tensor.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore.nn as nn
>>> import mindspore.ops as ops
>>>
>>>
>>> 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
...