eye 函数

public pure function eye(dim1, dim2) result(result)

构建单位矩阵。(规范)

参数

类型 意图可选 属性 名称
integer, intent(in) :: dim1
integer, intent(in), optional :: dim2

返回值 integer(kind=int8), allocatable, (:,:)


源代码

    pure function eye(dim1, dim2) result(result)

        integer, intent(in) :: dim1
        integer, intent(in), optional :: dim2
        integer(int8), allocatable :: result(:, :)

        integer :: dim2_
        integer :: i

        dim2_ = optval(dim2, dim1)
        allocate(result(dim1, dim2_))
        
        result = 0_int8
        do i = 1, min(dim1, dim2_)
            result(i, i) = 1_int8
        end do

    end function eye