Update real_time_encoder_transformer.py

Author: ajatshatru01

neural_network/real_time_encoder_transformer.py

@@ -38,8 +38,10 @@ def forward(self, time_steps: np.ndarray) -> np.ndarray:
         >>> out.shape
         (1, 3, 4)
         """
+      
+        
         linear = self.w0 * time_steps + self.b0
-        periodic = np.sin(self.w * time_steps + self.b)
+        periodic = np.sin(time_steps * self.w[:, None, :] + self.b[:, None, :])
         return np.concatenate([linear, periodic], axis=-1)