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

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

用法:

torch.distributed.Store.num_keys(self: torch._C._distributed_c10d.Store) → int

返回：

存储中存在的 key 数量。

返回存储中设置的键数。请注意，此数字通常比 set() 和 add() 添加的键数大 1，因为一个键用于协调使用存储的所有工作人员。

警告

与 TCPStore 一起使用时，num_keys 返回写入基础文件的键数。如果存储被破坏并使用同一文件创建另一个存储，则原始 key 将被保留。

例子：

>>> import torch.distributed as dist
>>> from datetime import timedelta
>>> # Using TCPStore as an example, other store types can also be used
>>> store = dist.TCPStore("127.0.0.1", 0, 1, True, timedelta(seconds=30))
>>> store.set("first_key", "first_value")
>>> # This should return 2
>>> store.num_keys()