Skip to main content

broadcast_binary

ruprim_host::expand

Function broadcast_binary 

Source 
pub fn broadcast_binary(
    lhs: HostTensor,
    rhs: HostTensor,
) -> (HostTensor, HostTensor)
Expand description

Broadcast two tensors to the same shape for binary operations.