丘比地图坐标给出的形状不匹配

Question

我试图使用scipy兼容的cupy函数，即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) 

现在，返回错误：ValueError: Out shape is mismatched。你知道吗

我现在不确定为什么会这样，因为输出应该由函数本身生成。坐标是正确的4D形状。我真不知道该怎么做才行。不幸的是，我在网上也找不到任何这样的例子。你知道吗

Answer 1

它看起来像丘比的一个bug（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:])