mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix typesetting issues

This commit is contained in:
zuochuanyong 2021-07-01 10:21:45 +08:00
parent 98f66ae3dd
commit ac9013caa8
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mindspore/ops/operations/nn_ops.py
View File

@ -148,14 +148,19 @@ class AdaptiveAvgPool2D(PrimitiveWithInfer):
Outputs:
Tensor, with the same type as the `input_x`.
Shape of the output is `input_x_shape[:len(input_x_shape) - len(out_shape)] + out_shape`.
If `output_size` contains `None`:
`out_shape = input_x_shape[-2] + output_size[1]`: If `output_size` is `(None, w)`
`out_shape = output_size[0] + input_x_shape[-1]`: If `output_size` is `(h, None)`
`out_shape = input_x_shape[-2:]: If output_size` is `(None, None)`
- `out_shape = input_x_shape[-2] + output_size[1]`: If `output_size` is `(None, w)`
- `out_shape = output_size[0] + input_x_shape[-1]`: If `output_size` is `(h, None)`
- `out_shape = input_x_shape[-2:]: If output_size` is `(None, None)`
If `output_size` does not contain `None`:
`out_shape = (h, h)`: If `output_size` is `h`
`out_shape = (h, w)`: If `output_size` is `(h, w)`
- `out_shape = (h, h)`: If `output_size` is `h`
- `out_shape = (h, w)`: If `output_size` is `(h, w)`
Raises:
ValueError: If `output_size` is a tuple and if `output_size` length is not 2.