mindspore.ParameterTuple

查看源文件
class mindspore.ParameterTuple[源代码]

继承于tuple,用于管理多个Parameter。

说明

该类把网络参数存储到参数元组集合中。

样例:

>>> from mindspore import Tensor, Parameter, ParameterTuple
>>> import numpy as np
>>> x = Parameter(Tensor(np.array([[1, 2], [3, 4]], dtype=np.float32)), name="param")
>>> y = Parameter(Tensor(np.array([[5, 6], [7, 8]], dtype=np.float32)), name="param1")
>>> pt = ParameterTuple([x, y])
>>> pt1 = pt.clone(prefix="new")
clone(prefix, init='same')[源代码]

逐个对ParameterTuple中的Parameter进行克隆,生成新的ParameterTuple。

参数:
  • prefix (str) - Parameter的namespace,此前缀将会被添加到Parametertuple中的Parameter的name属性中。

  • init (Union[Tensor, str, numbers.Number]) - 对Parametertuple中Parameter的shape和类型进行克隆,并根据传入的 init 设置数值。默认值: 'same'

    • 如果 initTensor ,则新参数的数值与该Tensor相同。

    • 如果 initnumbers.Number ,则设置新参数的数值为该值。

    • 如果 initstr ,则按照 Initializer 模块中对应的同名的初始化方法进行数值设定。若为 'same',则新参数的数值与原Parameter相同。

返回:

新的参数元组。

教程样例: