gluon docfix (#13631)

apache · Dec 29, 2018 · 9fbd7dd · 9fbd7dd
1 parent 991bf3b
commit 9fbd7dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/mxnet/gluon/data/vision/transforms.py b/python/mxnet/gluon/data/vision/transforms.py
@@ -82,10 +82,10 @@ class Cast(HybridBlock):
 
 
     Inputs:
-        - **data**: input tensor with arbitrary shape.
+        - **data**: input tensor with arbitrary shape and dtype.
 
     Outputs:
-        - **out**: output tensor with the same shape as `data`.
+        - **out**: output tensor with the same shape as `data` and data type as dtype.
     """
     def __init__(self, dtype='float32'):
         super(Cast, self).__init__()