summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py')
-rw-r--r--text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py b/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py
index 9466f6e..7140537 100644
--- a/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py
+++ b/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py
@@ -1,4 +1,5 @@
"""Absolute positional embedding."""
+import torch
from torch import nn, Tensor