私は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
が、これは何も変わりませんでした。