Skip to content

Commit 5111a87

Browse files
committed
bug fix unpackDistr PackedZeroMeanFullNormal
1 parent c1e06fb commit 5111a87

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/Serialization/services/SerializingDistributions.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@ function unpackDistribution(dtr::PackedZeroMeanDiagNormal)
2424
return MvNormal(LinearAlgebra.Diagonal(map(abs2, sqrt.(dtr.diag))))
2525
end # sqrt.(dtr.diag)
2626
function unpackDistribution(dtr::PackedZeroMeanFullNormal)
27-
return MvNormal(reshape(dtr.cov, length(dtr.mu), :))
27+
d = round(Int,sqrt(size(dtr.cov)[1]))
28+
return MvNormal(reshape(dtr.cov, d, d))
2829
end
2930
unpackDistribution(dtr::PackedDiagNormal) = MvNormal(dtr.mu, sqrt.(dtr.diag))
3031
function unpackDistribution(dtr::PackedFullNormal)

0 commit comments

Comments
 (0)