Update README.md

Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
David Rotermund 2023-12-22 14:38:49 +01:00 committed by GitHub
parent 4f05e0d255
commit 732d474a9c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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)