oᵖ = selectdim(ô, 1, c+1:size(ô, 1))
reduce(norm², oᵖ, einops"(d p) l h b -> p l h b"; d=3)
For some reason it's improperly unwrapping.
ERROR: DimensionMismatch: Split sizes incompatible with input size: input_prod=94 not divisible by known_prod=3
Stacktrace:
[1] macro expansion
@ ~/.julia/packages/Rewrap/Ge6Hx/src/Reshape/utils.jl:86 [inlined]
[2] split_sizes(op::Rewrap.Split1{2, Tuple{Int64, Colon}}, input_prod::Int64)
@ Rewrap ~/.julia/packages/Rewrap/Ge6Hx/src/Reshape/utils.jl:31
[3] macro expansion
@ ~/.julia/packages/Rewrap/Ge6Hx/src/Reshape/specializations/AbstractArray.jl:73 [inlined]
[4] (::Rewrap.Reshape{Tuple{Rewrap.Split1{2, Tuple{…}}, Rewrap.Keep{1}, Rewrap.Keep{1}, Rewrap.Keep{1}}, 4, 5})(x::Array{Float32, 4})
@ Rewrap ~/.julia/packages/Rewrap/Ge6Hx/src/Reshape/specializations/AbstractArray.jl:68
[5] macro expansion
@ ~/.julia/packages/Rewrap/Ge6Hx/src/Reshape/specializations/SubArray.jl:290 [inlined]
[6] (::Rewrap.Reshape{Tuple{…}, 4, 5})(x::SubArray{Float32, 4, Array{…}, Tuple{…}, false})
@ Rewrap ~/.julia/packages/Rewrap/Ge6Hx/src/Reshape/specializations/SubArray.jl:295
[7] reshape
@ ~/.julia/packages/Rewrap/Ge6Hx/src/enhanced-base/reshape.jl:42 [inlined]
[8] reshape
@ ~/.julia/packages/Rewrap/Ge6Hx/src/enhanced-base/reshape.jl:97 [inlined]
[9] macro expansion
@ ~/.julia/packages/Einops/C0pHx/src/rearrange.jl:38 [inlined]
[10] #rearrange#42
@ ~/.julia/packages/Einops/C0pHx/src/rearrange.jl:31 [inlined]
[11] (::Onion.FlashIPA{…})(s::Array{…}, z::Array{…}, ::Tuple{…}; kws::@Kwargs{})
@ Onion ~/speedrun/Onion/src/ipa/FlashIPA.jl:78
[12] (::Onion.FlashIPA{…})(s::Array{…}, z::Array{…}, ::Tuple{…})
@ Onion ~/speedrun/Onion/src/ipa/FlashIPA.jl:39
[13] top-level scope
@ REPL[9]:1
Some type information was truncated. Use `show(err)` to see complete types.
For some reason it's improperly unwrapping.