Tutorials
Examples
References
Other
Community
chainer.backend.
get_array_module
Gets an appropriate NumPy-compatible module to process arguments
This function will return their data arrays’ array module for Variable arguments.
Variable
args – Values to determine whether NumPy, CuPy, or ChainerX should be used.
numpy, cupy, or chainerx is returned based on the types of the arguments.
numpy
cupy
chainerx
module