We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9d3936a commit fd1d297Copy full SHA for fd1d297
python/mxnet/gluon/data/vision/transforms.py
@@ -112,7 +112,7 @@ class ToTensor(HybridBlock):
112
- **data**: input tensor with (H x W x C) or (N x H x W x C) shape and uint8 type.
113
114
Outputs:
115
- - **out**: output tensor with (C x H x W) or (N x H x W x C) shape and float32 type.
+ - **out**: output tensor with (C x H x W) or (N x C x H x W) shape and float32 type.
116
117
Examples
118
--------
0 commit comments