modify format
This commit is contained in:
parent
d79cfe29c5
commit
17a34c0e1e
|
@ -21,12 +21,12 @@ mindspore.ops.adaptive_avg_pool2d
|
||||||
* (w_{end}- w_{start})}
|
* (w_{end}- w_{start})}
|
||||||
\end{align}
|
\end{align}
|
||||||
|
|
||||||
**输入:**
|
**参数:**
|
||||||
|
|
||||||
- **input_x** (Tensor) - adaptive_avg_pool2d的输入,为三维或四维的Tensor,数据类型为float16、float32或者float64。
|
- **input_x** (Tensor) - adaptive_avg_pool2d的输入,为三维或四维的Tensor,数据类型为float16、float32或者float64。
|
||||||
- **output_size** (Union[int, tuple]) - 输出特征图的尺寸为H * W。可以是int类型的H和W组成的tuple,也可以是H * H的单个H,或None,如果是None,则意味着输出大小与输入相同。
|
- **output_size** (Union[int, tuple]) - 输出特征图的尺寸为H * W。可以是int类型的H和W组成的tuple,也可以是H * H的单个H,或None,如果是None,则意味着输出大小与输入相同。
|
||||||
|
|
||||||
**输出:**
|
**返回:**
|
||||||
|
|
||||||
Tensor,数据类型与 `input_x` 相同。
|
Tensor,数据类型与 `input_x` 相同。
|
||||||
|
|
||||||
|
|
|
@ -9,13 +9,13 @@ mindspore.ops.masked_fill
|
||||||
.. note::
|
.. note::
|
||||||
Ascend平台暂不支持batch维输入。即,`value` 要求为0维的Tensor或者float。
|
Ascend平台暂不支持batch维输入。即,`value` 要求为0维的Tensor或者float。
|
||||||
|
|
||||||
**输入:**
|
**参数:**
|
||||||
|
|
||||||
- **input_x** (Tensor) - 输入Tensor,其数据类型为float16、float32、int8、或int32。
|
- **input_x** (Tensor) - 输入Tensor,其数据类型为float16、float32、int8、或int32。
|
||||||
- **mask** (Tensor[bool]) - 输入的掩码,其数据类型为bool。
|
- **mask** (Tensor[bool]) - 输入的掩码,其数据类型为bool。
|
||||||
- **value** (Union[float, Tensor]) - 用来填充的值,其数据类型与 `input_x` 相同。
|
- **value** (Union[float, Tensor]) - 用来填充的值,其数据类型与 `input_x` 相同。
|
||||||
|
|
||||||
**输出:**
|
**返回:**
|
||||||
|
|
||||||
Tensor,输出与输入的数据类型和shape相同。
|
Tensor,输出与输入的数据类型和shape相同。
|
||||||
|
|
||||||
|
|
|
@ -5,7 +5,7 @@ mindspore.ops.matrix_band_part
|
||||||
|
|
||||||
将矩阵的每个中心带外的所有位置设置为0。
|
将矩阵的每个中心带外的所有位置设置为0。
|
||||||
|
|
||||||
`x` 、 `lower` 和 `upper` 三者的shapes必须相同或能够广播。
|
`x` 、 `lower` 和 `upper` 三者的shapes必须相同或能够广播。
|
||||||
|
|
||||||
**参数:**
|
**参数:**
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue