Function tch_utils::utils::assert_eq_tensor

pub fn assert_eq_tensor(a: &Tensor, b: &Tensor)