fixup I_A calc

Author: alyst

src/imply/RAM/generic.jl

@@ -224,12 +224,8 @@ function objective!(imply::RAM, parameters, model, has_meanstructure::Val{T}) wh
         parameters,
     )
 
-    @inbounds for (j, I_Aj, Aj) in
-                  zip(axes(imply.A, 2), eachcol(imply.I_A), eachcol(imply.A))
-        for i in axes(imply.A, 1)
-            I_Aj[i] = ifelse(i == j, 1, 0) - Aj[i]
-        end
-    end
+    @. I_A = -A
+    @view(I_A[diagind(I_A)]) .+= 1
 
     copyto!(imply.F⨉I_A⁻¹, imply.F)
     rdiv!(imply.F⨉I_A⁻¹, factorize(imply.I_A))
@@ -257,12 +253,8 @@ function gradient!(
         parameters,
     )
 
-    @inbounds for (j, I_Aj, Aj) in
-                  zip(axes(imply.A, 2), eachcol(imply.I_A), eachcol(imply.A))
-        for i in axes(imply.A, 1)
-            I_Aj[i] = ifelse(i == j, 1, 0) - Aj[i]
-        end
-    end
+    @. I_A = -A
+    @view(I_A[diagind(I_A)]) .+= 1
 
     imply.I_A⁻¹ = LinearAlgebra.inv!(factorize(imply.I_A))
     mul!(imply.F⨉I_A⁻¹, imply.F, imply.I_A⁻¹)