Function diagonal_mut 

pub fn diagonal_mut<R, T, B, D>(
    tensor: &mut TensorBase<Storage<R, T, B>, D>,
    diagonal_args: impl Into<DiagonalArgs>,
) -> TensorBase<Storage<DataMut<'_, <B as DeviceRawAPI<T>>::Raw>, T, B>, <D as DimSmallerOneAPI>::SmallerOne>
where
    D: DimAPI + DimSmallerOneAPI,
    <D as DimSmallerOneAPI>::SmallerOne: DimAPI,
    R: DataMutAPI<Data = <B as DeviceRawAPI<T>>::Raw>,
    B: DeviceAPI<T>,