load

class tensordict.load(prefix: str | pathlib.Path, device: Optional[device] = None, non_blocking: bool = False, *, out: Optional[TensorDictBase] = None)

从磁盘加载 tensordict。

此类方法是 load_memmap() 的代理。