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