reshape T

MurrellGroup · Apr 29, 2024 · 7c5d7f1 · 7c5d7f1
1 parent cc000ad
commit 7c5d7f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layers.jl b/src/layers.jl
@@ -247,7 +247,7 @@ function ipa_customgrad(ipa::Union{IPCrossA, IPA}, Ti::Tuple{AbstractArray,Abstr
     else
         use_softmax1 = false
     end
-    TiL = TiR = Ti 
+    TiL = TiR = (Ti[1], reshape(Ti[2], size(Ti[2],1), 1, size(Ti[2])[2:end]...)) 
     siL = siR = S
     rot_TiL, translate_TiL = TiL
     rot_TiR, translate_TiR = TiR