Python PyTorch Store.set用法及代码示例

本文简要介绍python语言中 torch.distributed.Store.set 的用法。

用法:

torch.distributed.Store.set(self: torch._C._distributed_c10d.Store, arg0: str, arg1: str) → None

参数：

• key(str) -要添加到商店的 key 。

• value(str) -与 key 关联的值将添加到存储中。

根据提供的 key 和 value 将键值对插入到存储中。如果 key 已存在于商店中，它将用新提供的 value 覆盖旧值。

例子：

>>> import torch.distributed as dist
>>> from datetime import timedelta
>>> store = dist.TCPStore("127.0.0.1", 0, 1, True, timedelta(seconds=30))
>>> store.set("first_key", "first_value")
>>> # Should return "first_value"
>>> store.get("first_key")