Fix cross masking

MurrellGroup · Feb 3, 2024 · 07a72ce · 07a72ce
1 parent ab36361
commit 07a72ce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layers.jl b/src/layers.jl
@@ -167,9 +167,9 @@ function (ipa::Union{IPCrossA, IPA})(TiL::Tuple{AbstractArray,AbstractArray}, si
 
     # Setting mask to the correct dim for broadcasting. 
     if mask != 0 
-        mask = reshape(mask, 1, N_frames_L, N_frames_R, 1) 
+        mask = reshape(mask, 1, N_frames_R, N_frames_L, 1) 
     end
-    
+
     att = Flux.softmax(w_L .* (att_arg .+ bij) .+ mask, dims = 3)
 
     # Applying the attention weights to the values.