Browse Source

I3BPK4: adjust description of input argument

tags/v1.2.0-rc1
zhuyuxiao 4 years ago
parent
commit
a11d17db1a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/timedistributed.py

+ 1
- 1
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, *)`.


Loading…
Cancel
Save