Skip to content

Commit

Permalink
docs(PCDDataset): add information for shape of y
Browse files Browse the repository at this point in the history
  • Loading branch information
adosar committed Jan 4, 2025
1 parent 59f26d0 commit 0187fc7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/aidsorb/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -333,7 +333,7 @@ class PCDDataset(Dataset):
.. note::
* ``x`` and ``y`` are tensors of ``dtype=torch.float``.
* ``y`` has shape ``(len(labels),)``.
* ``y`` has shape ``(len(labels),)`` if ``transform_y=None``.
* ``transform_x`` and ``transform_y`` expect :class:`~torch.Tensor` as
input.
Expand Down

0 comments on commit 0187fc7

Please sign in to comment.