diff --git a/python/mxnet/gluon/data/vision/transforms.py b/python/mxnet/gluon/data/vision/transforms.py index ab8f8ab482df..d2c7f906caac 100644 --- a/python/mxnet/gluon/data/vision/transforms.py +++ b/python/mxnet/gluon/data/vision/transforms.py @@ -112,7 +112,7 @@ class ToTensor(HybridBlock): - **data**: input tensor with (H x W x C) or (N x H x W x C) shape and uint8 type. Outputs: - - **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. Examples --------