Mojo module
broadcast
Functions
-
broadcast
: For each axis ofinput
, if the dimension is 1, duplicate the data at each index of the corresponding axis inoutput
, otherwise copy over the entire axis to the corresponding axis inoutput
. -
broadcast_impl
: For each axis ofinput
∈ [axis, rank), if the dimension is 1, duplicate the data at each index of the corresponding axis inoutput
, otherwise copy over the entire axis to the corresponding axis inoutput
.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!