funclp.use_broadcasting module

funclp.use_broadcasting(xp, variables, data, parameters, variables_shape, data_shape, parameters_shape, out_shape)

Defines inputs after broadcasting ready to use.

Parameters:

• xp (module) – numpy / cupy.

• variables (list) – List of input variables.

• data (list) – List of input data.

• parameters (list) – List of input parameters.

• variables_shape (tuple) – Shape to apply to variables.

• data_shape (tuple) – Shape to apply to data.

• parameters_shape (tuple) – Shape to apply to parameters.

• out_shape (tuple) – Output shape.

Returns:

• variables (list) – List of broadcasted variables.

• data (list) – List of broadcasted data.

• parameters (list) – List of broadcasted parameters.