mindspore.Tensor.fill_

查看源文件
mindspore.Tensor.fill_(value)

用指定的值填充 self

警告

这是一个实验性API,后续可能修改或删除。

参数:
  • value (Union[Tensor, number.Number, bool]) - 用来填充 self 的值。

返回:

Tensor。

异常:
  • RunTimeError - selfvalue 的数据类型不支持。

  • RunTimeError - 当 value 是Tensor时,它应该是0-D Tensor或shape=[1]的1-D Tensor。

支持平台:

Ascend

样例:

>>> import mindspore
>>> from mindspore import ops
>>> x = ops.zeros((3, 3))
>>> print(x)
[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
>>> output = x.fill_(1.0)
>>> print(output)
[[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]
>>> print(x)
[[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]