chainer.cuda.get_array_module(*args)[source]

numpy もしくは cupyから適当なものを取得する。

 

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

 

Parameters: args – NumPy もしくは CuPy を使用することを指定する値
Returns: cupy もしくは numpy が引数の型に基いて返される。
Return type: module