From 9fbd7dd26c2e49af36d78c1ea58d322f63a882c4 Mon Sep 17 00:00:00 2001 From: vdantu <36211508+vdantu@users.noreply.github.com> Date: Sat, 29 Dec 2018 14:17:12 -0500 Subject: [PATCH] gluon docfix (#13631) --- python/mxnet/gluon/data/vision/transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/mxnet/gluon/data/vision/transforms.py b/python/mxnet/gluon/data/vision/transforms.py index 3523be4d054a..175076925332 100644 --- 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__()