Function broadcast_in_dim 

pub fn broadcast_in_dim(
    input: &Tensor,
    shape: &[usize],
    dims: &[usize],
) -> Result<Tensor>