pub fn into_diagonal<S, D>(
tensor: TensorBase<S, D>,
diagonal_args: impl Into<DiagonalArgs>,
) -> TensorBase<S, D::SmallerOne>pub fn into_diagonal<S, D>(
tensor: TensorBase<S, D>,
diagonal_args: impl Into<DiagonalArgs>,
) -> TensorBase<S, D::SmallerOne>