mindspore.ops.ImageSummary

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

Outputs the image tensor to protocol buffer through image summary operator.

Inputs:

• name (str) - The name of the input variable, it must not be an empty string.

• value (Tensor) - The value of image, the rank of tensor must be 4.

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 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
...