espnet2.enh.diffusion.sdes.batch_broadcast

source

espnet2.enh.diffusion.sdes.batch_broadcast(a, x)

Broadcasts a over all dimensions of x, except the batch dimension,

which must match.