from_h5

class tensordict.from_h5(filename, mode='r')

從 h5 檔案建立 PersistentTensorDict。

此函數將自動確定每個巢狀 tensordict 的批次大小。

參數:

• filename (str) – h5 檔案的路徑。

• mode (str, optional) – 讀取模式。預設為 "r"。