0

私はcupy次の操作を行うために使用していますが、これはかなり高速です。

import cupy as cp

shape = (256, 170, 256)

deformation = cp.meshgrid(cp.arange(shape[0]),
                          cp.arange(shape[1]),
                          cp.arange(shape[2]),
                          indexing='ij')

ただし、次のように配列に変換すると:

deformation = cp.array(cp.meshgrid(cp.arange(shape[0]),
                                   cp.arange(shape[1]),
                                   cp.arange(shape[2]),
                                   indexing='ij'))

これは非常に遅いか、ハングするようです (5 分後にあきらめました)。ここで何が間違っているのかわかりません。

私も呼び出しに合格しようとしcopy=Falseましたcp.arrayが、これは何も変わりませんでした。

4

1 に答える 1