Skip to content

Commit 7f6a39d

Browse files
committed
clarify layout
1 parent 995ab7c commit 7f6a39d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

docs/source/transforms.rst

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,9 @@ Transforms tend to be sensitive to the input strides / memory layout. Some
156156
transforms will be faster with channels-first images while others prefer
157157
channels-last. You may want to experiment a bit if you're chasing the very
158158
best performance. Using :func:`torch.compile` on individual transforms may
159-
also help factoring out the layout variable (e.g. on
160-
:class:`~torchvision.transforms.v2.Normalize`).
159+
also help factoring out the memory layout variable (e.g. on
160+
:class:`~torchvision.transforms.v2.Normalize`). Note that we're talking about
161+
**memory layout**, not tensor shape.
161162

162163
Note that resize transforms like :class:`~torchvision.transforms.v2.Resize`
163164
and :class:`~torchvision.transforms.v2.RandomResizedCrop` typically prefer

0 commit comments

Comments
 (0)