Python PyTorch Store.compare_set用法及代碼示例

本文簡要介紹python語言中 torch.distributed.Store.compare_set 的用法。

用法:

torch.distributed.Store.compare_set(self: torch._C._distributed_c10d.Store, arg0: str, arg1: str, arg2: str) → bytes

參數：

• key(str) -要在商店中檢查的 key 。

• expected_value(str) -與key 關聯的值在插入前被檢查。

• desired_value(str) -與 key 關聯的值將添加到存儲中。

根據提供的 key 將鍵值對插入到 store 中，並在插入前對 expected_value 和 desired_value 進行比較。僅當key 的expected_value 已存在於商店中或expected_value 為空字符串時，才會設置desired_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("key", "first_value")
>>> store.compare_set("key", "first_value", "second_value")
>>> # Should return "second_value"
>>> store.get("key")