Skip to content

Commit

Permalink
add float16 to pixel_shuffle_cn (PaddlePaddle#5676)
Browse files Browse the repository at this point in the history
Co-authored-by: wqgo <[email protected]>
  • Loading branch information
Bjmw3 and ShiYue-oo authored Mar 6, 2023
1 parent 9a37f65 commit 1fd1e83
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/api/paddle/nn/functional/pixel_shuffle_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ pixel_shuffle

参数
:::::::::
- **x** (Tensor) - 当前算子的输入,其是一个形状为 `[N, C, H, W]` 的 4-D Tensor。其中 `N` 是 batch size, `C` 是通道数,`H` 是输入特征的高度,`W` 是输入特征的宽度。其数据类型为 float32 或者 float64。
- **x** (Tensor) - 当前算子的输入,其是一个形状为 `[N, C, H, W]` 的 4-D Tensor。其中 `N` 是 batch size, `C` 是通道数,`H` 是输入特征的高度,`W` 是输入特征的宽度。其数据类型为 float16,float32float64。
- **upscale_factor** (int) - 增大空间分辨率的增大因子
- **data_format** (str,可选) - 数据格式,可选:"NCHW"或"NHWC",默认:"NCHW"
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
Expand Down

0 comments on commit 1fd1e83

Please sign in to comment.