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")