lib.utils.enforce_shape¶

lib.utils.enforce_shape(x, y, grid=True)¶

Broadcast x and y arrays.

Parameters

x (array_like) – Input x-array, scalar, 1D or 2D if not grid.
y (array_like) – Input y-array, scalar, 1D or 2D if not grid.
grid (bool, default=True) – If grid, and x, y not scalars, add dimension to x such that x and y can be broadcast (e.g. x*y, is shape (len(x),) + y.shape). Else, simply return x,y.

Returns