快捷方式

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
clear()[source]

删除 ParameterDict 中的所有项。

copy()[source]

返回此 ParameterDict 实例的副本。

返回类型

ParameterDict

fromkeys(keys, default=None)[source]

返回一个新的 ParameterDict,其中包含提供的键。

参数
  • keys (iterable, string) – 用于创建新 ParameterDict 的键

  • default (Parameter, 可选) – 为所有键设置的值

返回类型

ParameterDict

get(key, default=None)[source]

如果存在,则返回与键关联的参数。否则,如果提供了默认值,则返回默认值,否则返回 None。

参数
  • key (str) – 要从 ParameterDict 中获取的键

  • default (Parameter, 可选) – 如果键不存在,则返回的值

返回类型

任意类型

items()[source]

返回 ParameterDict 键/值对的可迭代对象。

返回类型

Iterable[Tuple[str, Any]]

keys()[source]

返回 ParameterDict 的键的可迭代对象。

返回类型

Iterable[str]

pop(key)[source]

从 ParameterDict 中移除键并返回其参数。

参数

key (str) – 要从 ParameterDict 中弹出的键

返回类型

任意类型

popitem()[source]

移除并返回 ParameterDict 中最后插入的(key, parameter)对。

返回类型

Tuple[str, Any]

setdefault(key, default=None)[source]

设置 ParameterDict 中键的默认值。

如果键在 ParameterDict 中,则返回其值。否则,插入键及其参数default并返回default。default默认为None。

参数
  • key (str) – 要设置默认值的键

  • default (Any) – 设置给键的参数

返回类型

任意类型

update(parameters)[source]

使用来自parameters的键值对更新ParameterDict,覆盖现有键。

注意

如果parametersOrderedDictParameterDict或键值对的可迭代对象,则保留其中新元素的顺序。

参数

parameters (iterable) – 从字符串到Parameter的映射(字典),或类型为(字符串,Parameter)的键值对的可迭代对象。

values()[source]

返回 ParameterDict 值的可迭代对象。

返回类型

Iterable[Any]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源