ParameterDict¶
- class torch.nn.ParameterDict(parameters=None)[source]¶
将参数保存在字典中。
ParameterDict 可以像常规的 Python 字典一样索引,但它包含的参数已正确注册,并且所有 Module 方法都可以看到它们。其他对象将像常规 Python 字典一样处理。
ParameterDict
是一个**有序**字典。使用其他无序映射类型(例如 Python 的普通dict
)进行update()
不会保留合并映射的顺序。另一方面,OrderedDict
或另一个ParameterDict
将保留其顺序。请注意,构造函数、字典元素的赋值和
update()
方法会将任何Tensor
转换为Parameter
。- 参数
values (iterable, 可选) – (字符串 : 任意类型) 的映射(字典)或 (字符串, 任意类型) 类型键值对的可迭代对象
示例
class MyModule(nn.Module): def __init__(self) -> None: super().__init__() self.params = nn.ParameterDict({ 'left': nn.Parameter(torch.randn(5, 10)), 'right': nn.Parameter(torch.randn(5, 10)) }) def forward(self, x, choice): x = self.params[choice].mm(x) return x
- copy()[source]¶
返回此
ParameterDict
实例的副本。- 返回类型
- fromkeys(keys, default=None)[source]¶
返回一个新的 ParameterDict,其中包含提供的键。
- 参数
keys (iterable, string) – 用于创建新 ParameterDict 的键
default (Parameter, 可选) – 为所有键设置的值
- 返回类型
- setdefault(key, default=None)[source]¶
设置 ParameterDict 中键的默认值。
如果键在 ParameterDict 中,则返回其值。否则,插入键及其参数default并返回default。default默认为None。
- update(parameters)[source]¶
使用来自
parameters
的键值对更新ParameterDict
,覆盖现有键。注意
如果
parameters
是OrderedDict
、ParameterDict
或键值对的可迭代对象,则保留其中新元素的顺序。- 参数
parameters (iterable) – 从字符串到
Parameter
的映射(字典),或类型为(字符串,Parameter
)的键值对的可迭代对象。