From 732d474a9cebe34918f56141618a0a264408a1a2 Mon Sep 17 00:00:00 2001 From: David Rotermund <54365609+davrot@users.noreply.github.com> Date: Fri, 22 Dec 2023 14:38:49 +0100 Subject: [PATCH] Update README.md Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com> --- pytorch/interfacing_data/README.md | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/pytorch/interfacing_data/README.md b/pytorch/interfacing_data/README.md index 2fb5469..48458d9 100644 --- a/pytorch/interfacing_data/README.md +++ b/pytorch/interfacing_data/README.md @@ -14,6 +14,17 @@ Questions to [David Rotermund](mailto:davrot@uni-bremen.de) There are options to interface your data. +## [torch.utils.data.TensorDataset](https://pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset) + +```python +CLASS torch.utils.data.TensorDataset(*tensors) +``` + +> Dataset wrapping tensors. +> +> Each sample will be retrieved by indexing tensors along the first dimension. +> +> **\*tensors** : (Tensor) – tensors that have the same size of the first dimension. ## [torch.utils.data.Dataset](https://pytorch.org/docs/stable/data.html#torch.utils.data.Dataset)