mindspore.Tensor.set_const_arg

查看源文件
mindspore.Tensor.set_const_arg(const_arg=True)[源代码]

指定该Tensor在作为网络入参时是否是一个常量。

参数:
  • const_arg (bool) - Tensor在作为网络入参时是否是一个常量。默认值: True

返回:

Tensor,被指定了是否是一个常量网络入参。

异常:
  • TypeError - 如果 const_arg 不是一个布尔值。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1,2,3],[4,5,6]], dtype=np.float32))
>>> x.set_const_arg(True)