torch_frame.utils.save

save(tensor_frame: TensorFrame, col_stats: dict[str, dict[torch_frame.data.stats.StatType, Any]] | None, path: str)[source]

Save a TensorFrame object and optional col_stats to specified path.

Parameters:

  • tensor_frame (TensorFrame) – The TensorFrame object to be saved.

  • col_stats (dict, optional) – The col_stats to be saved.

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