ListStorage

class torchrl.data.replay_buffers.ListStorage(max_size: int | None = None, *, compilable: bool = False, device: torch.device | str | int | None = None)

一個儲存在列表中的儲存器。

此類無法與 PyTrees 擴充套件，在呼叫 extend() 時提供的資料應為可迭代物件（如列表、元組、張量或具有非空批次大小的 TensorDicts）。

引數:

max_size (int, optional) – 儲存在儲存器中的最大元素數量。如果未提供，則建立一個無限制的儲存器。

關鍵字引數:

• compilable (bool, optional) – 如果為 True，儲存器將與 compile() 相容，但代價是在多程序環境中執行。

• device (str, optional) – 用於儲存器的裝置。預設為 None（輸入不移動到裝置）。

attach(buffer: Any) → None

此函式將取樣器附加到此儲存。

從該儲存讀取的緩衝區必須透過呼叫此方法作為已附加實體包含進來。這確保了當儲存中的資料發生變化時，元件能夠感知到這些變化，即使該儲存與其他緩衝區（例如，Priority Samplers）共享。

引數:

buffer – 讀取此儲存的物件。

dump(*args, **kwargs)

dumps() 的別名。

load(*args, **kwargs)

loads() 的別名。

save(*args, **kwargs)

dumps() 的別名。