pub fn diagonal<R, T, B, D>(
tensor: &TensorBase<Storage<R, T, B>, D>,
diagonal_args: impl Into<DiagonalArgs>,
) -> TensorBase<Storage<DataRef<'_, <B as DeviceRawAPI<T>>::Raw>, T, B>, <D as DimSmallerOneAPI>::SmallerOne>pub fn diagonal<R, T, B, D>(
tensor: &TensorBase<Storage<R, T, B>, D>,
diagonal_args: impl Into<DiagonalArgs>,
) -> TensorBase<Storage<DataRef<'_, <B as DeviceRawAPI<T>>::Raw>, T, B>, <D as DimSmallerOneAPI>::SmallerOne>