gluon docfix (apache#13631)

rondogency · Jan 9, 2019 · 18cdda2 · 18cdda2
1 parent add5462
commit 18cdda2
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__()