File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -156,8 +156,9 @@ Transforms tend to be sensitive to the input strides / memory layout. Some
156156transforms will be faster with channels-first images while others prefer
157157channels-last. You may want to experiment a bit if you're chasing the very
158158best 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
162163Note that resize transforms like :class: `~torchvision.transforms.v2.Resize `
163164and :class: `~torchvision.transforms.v2.RandomResizedCrop ` typically prefer
You can’t perform that action at this time.
0 commit comments