torch_frame.utils.load

load(path: str, device: torch.device | None = None) → tuple[TensorFrame, dict[str, dict[StatType, Any]] | None][source]

从指定路径加载保存的 TensorFrame 对象和可选的 col_stats。

Parameters:

  • path (str) – 用于加载 TensorFrame 对象和 col_stats 的路径。

  • device (torch.device, optional) – 用于加载 TensorFrame 对象的设备。（默认值：None）

Returns:

一个包含加载的 TensorFrame 对象的元组和

可选的 col_stats。

Return type:

tuple