torch_frame.data — pytorch-frame documentation (original) (raw)
Contents
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. |
|---|