p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!13537 TimeDistributed: change description of input argument 

From: @zyx5256
Reviewed-by: @wuxuejian,@liangchenghui
Signed-off-by: @wuxuejian
This commit is contained in:
mindspore-ci-bot 2021-03-19 11:23:59 +08:00 committed by Gitee
parent 9a4ccaf913 a11d17db1a
commit caaff9ab03
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/layer/timedistributed.py
View File

@ -70,7 +70,7 @@ class TimeDistributed(Cell):
Args: Args:
layer(Union[Cell, Primitive]): The Cell or Primitive which will be wrapped. layer(Union[Cell, Primitive]): The Cell or Primitive which will be wrapped.
time_axis(int): The axis of time_step. 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: Inputs:
- **input** (Tensor) - Tensor of shape :math:`(N, T, *)`. - **input** (Tensor) - Tensor of shape :math:`(N, T, *)`.