From a11d17db1a713ba92b6eec44c24f5ccfc08e17cc Mon Sep 17 00:00:00 2001 From: zhuyuxiao Date: Thu, 18 Mar 2021 15:50:34 +0800 Subject: [PATCH] I3BPK4: adjust description of input argument --- mindspore/nn/layer/timedistributed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/layer/timedistributed.py b/mindspore/nn/layer/timedistributed.py index 8f323ba1d9..f5ba48e917 100644 --- a/mindspore/nn/layer/timedistributed.py +++ b/mindspore/nn/layer/timedistributed.py @@ -70,7 +70,7 @@ class TimeDistributed(Cell): Args: layer(Union[Cell, Primitive]): The Cell or Primitive which will be wrapped. time_axis(int): The axis of time_step. - reshape_with_axis(int): The axis which time_axis will be reshaped with. Default: None. + reshape_with_axis(int): The axis which will be reshaped with time_axis. Default: None. Inputs: - **input** (Tensor) - Tensor of shape :math:`(N, T, *)`.