torch_frame.data

Data Objects

TensorFrame	A tensor frame holds a PyTorch tensor for each table column.
MultiEmbeddingTensor	A read-only PyTorch tensor-based data structure that stores [num_rows, num_cols, *], where the size of last dimension can be different for different column.
MultiNestedTensor	A read-only PyTorch tensor-based data structure that stores [num_rows, num_cols, *], where the size of last dimension can be different for different row/column.
Dataset	A base class for creating tabular datasets.

Stats

StatType

The different types for column statistics.

Data Loaders

DataLoader

A data loader which creates mini-batches from a torch_frame.Dataset or torch_frame.TensorFrame object.

Helper Functions

download_url	Downloads the content of url to the specified folder root.
DataFrameToTensorFrameConverter	A data frame to TensorFrame converter.