chainerx.atleast_2d

chainerx.atleast_2d(a)

View inputs as arrays with at least two dimensions.

Parameters

a (ndarray) – Array.

Returns

An array with a.ndim >= 2. Copies are avoided where possible, and views with two or more dimensions are returned.

Return type

ndarray

Note

  • Arrays that already have two or more dimensions are preserved.

  • During backpropagation, this function propagates the gradient of the output array to the input arrays in a.