Update real_time_encoder_transformer.py

ajatshatru01 · web-flow · commit 0974fee722b5 · 2025-10-21T21:55:07.000+05:30
diff --git a/neural_network/real_time_encoder_transformer.py b/neural_network/real_time_encoder_transformer.py
@@ -1,7 +1,8 @@
 # imports
 import math
+
 import torch
-from torch import nn
+from torch import nn, Tensor
 
 
 class Time2Vec(nn.Module):
