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)