There was an error while loading. Please reload this page.
1 parent bfd2bfa commit 468a024Copy full SHA for 468a024
1 file changed
neural_network/sliding_window_attention.py
@@ -68,7 +68,7 @@ def forward(self, input_tensor: np.ndarray) -> np.ndarray:
68
>>> (output.sum() != 0).item() # Check if output is non-zero
69
True
70
"""
71
- batch_size, seq_length, _ = input_tensor.shape
+ _batch_size, seq_length, _ = input_tensor.shape
72
output = np.zeros_like(input_tensor)
73
74
for i in range(seq_length):
0 commit comments