jax.lax.broadcast_to_rank

Contents

jax.lax.broadcast_to_rank#

jax.lax.broadcast_to_rank(x, rank)[source]#

Adds leading dimensions of 1 to give x rank rank.

Parameters:

x (Array) –
rank (int) –

Return type: