• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1mindspore.ops.FractionalMaxPool3DWithFixedKsize
2===============================================
3
4.. py:class:: mindspore.ops.FractionalMaxPool3DWithFixedKsize(ksize, output_shape, data_format="NCDHW")
5
6    此运算对由多个输入平面组成的输入信号进行3D分数最大池化。
7    最大池化操作在 :math:`(kD, kH, kW)` 区域中进行。其步长是随机的，大小由 `output_shape` 决定。
8
9    输出特征的数量等于输入平面的数量。
10
11    详细内容请参考论文 `Fractional MaxPooling by Ben Graham <https://arxiv.org/abs/1412.6071>`_ 。
12
13    输入和输出的数据格式可以是"NCDHW"和"NDHWC"，N是批量大小，C是通道数，D是特征深度，H是特征高度，W是特征宽度。
14
15    .. warning::
16        这是一个实验性API，后续可能修改或删除。
17
18    参数：
19        - **ksize** (Union[float, tuple]) - 池化窗口大小。可以是一个三元组，表示shape为 :math:`(k_D, k_H, k_W)` ，或者是一个单独的 `K` 表示shape为 :math:`(K, K, K)` 。
20        - **output_shape** (Union[int, tuple]) - 目标输出shape，可以是一个三元组，表示shape为 :math:`(D_{out}, H_{out}, W_{out})` ，或者是一个单独的 `S` 表示shape为 :math:`(S, S, S)` 。
21        - **data_format** (str，可选) - 可选的数据格式值，当前支持 ``"NCDHW"`` 和 ``"NHDWC"`` ，默认为 ``"NCDHW"`` 。
22
23    输入：
24        - **x** (Tensor) - 输入Tensor，是一个4-D或者5-D的Tensor。数据类型为：float16、float32、double、int32、int64。
25          支持的shape :math:`(N, C, D_{in}, H_{in}, W_{in})` 或者 :math:`(N, D_{in}, H_{in}, W_{in}, C)`。
26
27        - **random_samples** (Tensor) - 随机步长，是一个3-D Tensor，数据类型为：float16、float32、double，值在(0, 1)之间。支持的shape为 :math:`(N, C, 3)`。
28
29    输出：
30        - **y** (Tensor) - 一个Tensor，与 `x` 具有相同的dtype，shape为 :math:`(N, C, D_{out}, H_{out}, W_{out})`
31          或者 :math:`(N, D_{out}, H_{out}, W_{out}, C)`。
32
33        - **argmax** (Tensor) - 一个Tensor，输出的索引值。与 `y` 具有相同的shape，dype为int32或者int64。
34
35    异常：
36        - **TypeError** - 如果 `input_x` 不是一个4-D或者5-D的Tensor。
37        - **TypeError** - 如果 `random_samples` 不是一个3-D的Tensor。
38        - **TypeError** - 如果 `x` 的数据类型不是：float16、float32、double、int32、int64。
39        - **TypeError** - 如果 `random_samples` 的数据类型不是：float16、float32、double。
40        - **TypeError** - 如果 `argmax` 不是int32、int64。
41        - **ValueError** - 如果 `output_shape` 不是一个元组，并且 `output_shape` 长度不是3。
42        - **ValueError** - 如果 `ksize` 是一个元组，并且 `ksize` 长度不是3。
43        - **ValueError** - 如果 `output_shape` 或者 `ksize` 中的数值不是正数。
44        - **ValueError** - 如果 `data_format` 不是'NCDHW'，也不是'NDHWC'。
45        - **ValueError** - 如果 `input_x` 和 `random_samples` 的第一维大小不相等。
46        - **ValueError** - 如果 `input_x` and `random_samples` 的第二维大小不相等。
47        - **ValueError** - 如果 `random_samples` 的第三维大小不是3。
48