Function atg_diagonal 

pub unsafe extern "C" fn atg_diagonal(
    out__: *mut *mut C_tensor,
    self_: *mut C_tensor,
    offset_: i64,
    dim1_: i64,
    dim2_: i64,
)