cupy.expand_dims

cupy.expand_dims(a, axis)[source]

Expands given arrays.

Parameters
  • a (cupy.ndarray) – Array to be expanded.

  • axis (int) – Position where new axis is to be inserted.

Returns

The number of dimensions is one greater than that of

the input array.

Return type

cupy.ndarray