cupy map_coordinates 给出形状不匹配的错误

Question

我正在尝试使用 cupy 的 scipy 兼容功能，即 map_coordinates 功能。但是，经过大量修补后，我无法让它工作。所以，我做了如下事情：

import cupy as cp
import cupyx
import cupyx.scipy.ndimage
import numpy as np

# Generate some random data
x = np.random.rand(10, 10, 10)
# Move it to the GPU
x = cp.array(x)

# Generate some coordinates
d = [cp.array(np.random.rand(10, 10, 10)), 
     cp.array(np.random.rand(10, 10, 10)), 
     cp.array(np.random.rand(10, 10, 10))]

d = cp.stack(d)

print(d.shape)   # (3, 10, 10, 10)

cupyx.scipy.ndimage.map_coordinates(x, d, cp.float32, order=1, 
                                    mode='constant', cval=0)

现在，这个 returns 错误：ValueError: Out shape is mismatched。

我现在不确定为什么会这样，因为输出应该由函数本身生成。想想看，坐标是正确的 4D 形状。我真的不知道如何进行这项工作。不幸的是，我在网上也找不到任何这样的工作示例。

Answer 1

这看起来像是 CuPy（v6 和 v7 beta2）的一个错误。我提交了 issue。它仅在输出具有多个轴时失败，因此在错误修复之前，您可以通过展平除第一个轴以外的坐标数组来解决此错误，例如：

cupyx.scipy.ndimage.map_coordinates(
    x, d.reshape(len(d), -1), order=1, mode='constant', cval=0
).reshape(d.shape[1:])

cupy map_coordinates 给出形状不匹配的错误

cupy map_coordinates gives out shape is mismatched error

python

scipy

chainer

cupy