broadcast_batch_dims

broadcast_batch_dims#

Signature#

nabla.broadcast_batch_dims(arg: nabla.core.array.Array, batch_dims: tuple[int, ...]) -> nabla.core.array.Array

Description#

Broadcast array to target batch_dims.