supportdim

cscherrer · cscherrer · commit 917d92014ea9 · 2021-10-18T09:36:57.000-07:00
diff --git a/src/combinators/affine.jl b/src/combinators/affine.jl
@@ -224,3 +224,9 @@ end
 #     apply!(x, f, z)
 #     return z
 # end
+
+supportdim(nt::NamedTuple{(:μ,:σ)}) = colsize(nt.σ)
+supportdim(nt::NamedTuple{(:μ,:ω)}) = rowsize(nt.ω)
+supportdim(nt::NamedTuple{(:σ,)})   = colsize(nt.σ)
+supportdim(nt::NamedTuple{(:ω,)})   = rowsize(nt.ω)
+supportdim(nt::NamedTuple{(:μ,)})   = size(nt.μ)
