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