Update README.md
Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
parent
4f05e0d255
commit
732d474a9c
1 changed files with 11 additions and 0 deletions
|
@ -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)
|
||||
|
|
Loading…
Reference in a new issue