load

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

从磁盘加载 tensordict。

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