Cupy map_coordinates 给出形状不匹配错误

Posted 2023-02-16

【中文标题】Cupy map_coordinates 给出形状不匹配错误

【英文标题】：cupy map_coordinates gives out shape is mismatched error

【发布时间】：2019-07-29 20:43:23

【问题描述】：

我正在尝试使用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) 

现在，这会返回错误：ValueError: Out shape is mismatched。

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

【参考方案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:])