torch_frame.utils.load

load(path: str, device: Optional[device] = None) → tuple[torch_frame.data.tensor_frame.TensorFrame, dict[str, dict[torch_frame.data.stats.StatType, Any]] | None][source]

Load saved TensorFrame object and optional col_stats from a specified path.

Parameters:

  • path (str) – Path to load the TensorFrame object and col_stats.

  • device (torch.device, optional) – Device to load the TensorFrame object. (default: None)

Returns:

A tuple of loaded TensorFrame object and

optional col_stats.

Return type:

tuple