Update highwaylayer.md

kyegomez · Dec 27, 2023 · 12e00c4 · 12e00c4
1 parent e54e15d
commit 12e00c4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/zeta/nn/modules/highwaylayer.md b/docs/zeta/nn/modules/highwaylayer.md
@@ -51,6 +51,8 @@ Returns:
 ```python
 import torch.nn as nn
 import torch.nn.functional as F
+from zeta.nn import HighwayLayer
+
 
 class HighwayLayer(nn.Module):
  def __init__(self, dim):