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.