mindspore.ops.FractionalMaxPool3DWithFixedKsize
- class mindspore.ops.FractionalMaxPool3DWithFixedKsize(ksize, output_shape, data_format='NCDHW')[源代码]
此运算对由多个输入平面组成的输入信号进行3D分数最大池化。 最大池化操作在 \((kD, kH, kW)\) 区域中进行。其步长是随机的,大小由 output_shape 决定。
输出特征的数量等于输入平面的数量。
详细内容请参考论文 Fractional MaxPooling by Ben Graham 。
输入和输出的数据格式可以是”NCDHW”和”NDHWC”,N是批量大小,C是通道数,D是特征深度,H是特征高度,W是特征宽度。
警告
这是一个实验性API,后续可能修改或删除。
- 参数:
ksize (Union[float, tuple]) - 池化窗口大小。可以是一个三元组,表示shape为 \((k_D, k_H, k_W)\) ,或者是一个单独的 K 表示shape为 \((K, K, K)\) 。
output_shape (Union[int, tuple]) - 目标输出shape,可以是一个三元组,表示shape为 \((D_{out}, H_{out}, W_{out})\) ,或者是一个单独的 S 表示shape为 \((S, S, S)\) 。
data_format (str,可选) - 可选的数据格式值,当前支持
"NCDHW"
和"NHDWC"
,默认为"NCDHW"
。
- 输入:
x (Tensor) - 输入Tensor,是一个4-D或者5-D的Tensor。数据类型为:float16、float32、double、int32、int64。 支持的shape \((N, C, D_{in}, H_{in}, W_{in})\) 或者 \((N, D_{in}, H_{in}, W_{in}, C)\)。
random_samples (Tensor) - 随机步长,是一个3-D Tensor,数据类型为:float16、float32、double,值在(0, 1)之间。支持的shape为 \((N, C, 3)\)。
- 输出:
y (Tensor) - 一个Tensor,与 x 具有相同的dtype,shape为 \((N, C, D_{out}, H_{out}, W_{out})\) 或者 \((N, D_{out}, H_{out}, W_{out}, C)\)。
argmax (Tensor) - 一个Tensor,输出的索引值。与 y 具有相同的shape,dype为int32或者int64。
- 异常:
TypeError - 如果 input_x 不是一个4-D或者5-D的Tensor。
TypeError - 如果 random_samples 不是一个3-D的Tensor。
TypeError - 如果 x 的数据类型不是:float16、float32、double、int32、int64。
TypeError - 如果 random_samples 的数据类型不是:float16、float32、double。
TypeError - 如果 argmax 不是int32、int64。
ValueError - 如果 output_shape 不是一个元组,并且 output_shape 长度不是3。
ValueError - 如果 ksize 是一个元组,并且 ksize 长度不是3。
ValueError - 如果 output_shape 或者 ksize 中的数值不是正数。
ValueError - 如果 data_format 不是’NCDHW’,也不是’NDHWC’。
ValueError - 如果 input_x 和 random_samples 的第一维大小不相等。
ValueError - 如果 input_x and random_samples 的第二维大小不相等。
ValueError - 如果 random_samples 的第三维大小不是3。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> x = Tensor(np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]) ... .reshape([1, 1, 2, 2, 4]), mstype.float32) >>> random_samples = Tensor(np.array([0.7, 0.7, 0.7]).reshape([1, 1, 3]), mstype.float32) >>> ksize = (1, 1, 1) >>> output_shape = (1, 1, 2) >>> net = ops.FractionalMaxPool3DWithFixedKsize(ksize = ksize, output_shape = output_shape) >>> output, argmax = net(x, random_samples) >>> print(output) [[[[[13. 16.]]]]] >>> print(argmax) [[[[[12 15]]]]]