chainer.cuda.
get_array_module
numpy もしくは cupyから適当なものを取得する。
numpy
cupy
この関数は、 cupy.get_array_module()とほぼ同様です。異なる点は、この関数はCUDAが利用不可能な場合も使用することが可能であることと、 Variable 引数のためのデータ配列の配列モジュールを返すことです。
cupy.get_array_module()
Variable