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