mindspore.ops.FractionalMaxPool

class mindspore.ops.FractionalMaxPool(pooling_ratio, pseudo_random=False, overlapping=False, deterministic=False, seed=0, seed2=0)[源代码]

在输入上执行分数最大池化。

分数最大池化类似于常规最大池化。在常规最大池中,通过获取集合中较小N×N子部分的最大值(通常为2x2)来缩小输入集的大小,并尝试将集合减少N倍,其中N是整数。

分数最大池化意味着整体缩减比率N不必是整数。池区域的大小是随机生成的,但是相当均匀。

警告

pooling_ratio 当前只支持行和列轴,并行大于1.0,第一个和最后一个元素必须为1.0,因为我们不允许对batch和通道轴进行池化。

参数:
  • pooling_ratio (list(float)) - 决定了输出的shape,floats列表,长度大于等于4。对每个轴的value应该大于等于0,目前仅支持行和列维度。 第一个和最后一个元素必须为1.0,因为我们不允许对batch和通道轴进行池化。

  • pseudo_random (bool,可选) - 当设置为True时,以伪随机方式生成池序列,否则以随机方式生成池序列。默认为False。 查看文章 Fractional Max-Pooling 以了解伪随机和随机之间的差异。

  • overlapping (bool,可选) - 当设置为True时,表示池化时,两个单元格都使用相邻池化单元边界的值, 设置为False时,表示值不进行重复使用。默认为False。

  • deterministic (bool,可选) - 当设置为True时,将在计算图中的FractionalMaxPool节点上进行迭代时使用固定池区域。 主要用于单元测试,使FractionalMaxPool具有确定性。当设置为False时,将不使用固定池区域。默认为False。

  • seed (int,可选) - 如果seed或seed2被设置为非零,则随机数生成器由给定的seed生成,否则,它由随机种子生成。默认为0。

  • seed2 (int,可选) - 第二个seed,以避免发生seed碰撞。默认为0。

输入:
  • x (Tensor) - 数据类型必须为:float32、float64、int32、int64。shape为: \((N, H_{in}, W_{in}, C_{in})\)

输出:
  • y (Tensor) - 一个Tensor,FractionalMaxPool的输出,与 x 具有相同的数据类型,shape为: \((N, H_{out}, W_{out}, C_{out})\)

  • row_pooling_sequence (Tensor) - 一个Tensor,池边界行的结果列表,数据类型为int64。

  • col_pooling_sequence (Tensor) - 一个Tensor,池边界列的结果列表,数据类型为int64。

异常:
  • TypeError - 如果 x 数据类型不是:float32、float64、int32或者int64。

  • TypeError - 如果 x 不是一个4D的Tensor。

  • ValueError - 如果 x 的元素等于0或者小于0。

  • ValueError - 如果 pooling_ratio 是一个列表,其长度不等于4。

  • ValueError - 如果 pooling_ratio 的第一个和最后一个值不等于1.0。

支持平台:

GPU CPU

样例:

>>> x = np.array([1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]).reshape([1,4,4,1]).astype(np.int64)
>>> pooling_ratio=[1.0,1.5,1.5,1.0]
>>> fractionalmaxpool_op = ops.FractionalMaxPool(pooling_ratio=pooling_ratio)
>>> output = fractionalmaxpool_op(Tensor(x))
>>> print(output)
(Tensor(shape=[1, 2, 2, 1], dtype=Int64, value=
[[[[ 6],
   [ 8]],
  [[14],
   [16]]]]), Tensor(shape=[3], dtype=Int64, value= [0, 2, 4]), Tensor(shape=[3], dtype=Int64, value= [0, 2, 4]))