paddle.fluid.layers.unfold op不支持输入feature的shape存在多个-1
Created by: rfdeng
paddle.fluid.layers.unfold(x, kernel_size, strides=1, paddings=0, dilation=1, name=None)
unfolding层的第一个输入参数x,其shape不支持多个-1的描述方式。
对应代码及报错:
第一次调用unfold时,对应输入的shape为 (-1, 6, 1024, 768)是没问题的。
第二次调用unfold时,对应输入的shape为(-1, 16, -1, -1)报错。