Method forward

lam/lam/modules/blocks.py:37–38 · view source on GitHub ↗

(self, x: Tensor)

Source from the content-addressed store, hash-verified

35	self.pos_enc = pe
36
37	def forward(self, x: Tensor) -> Tensor:
38	return x + self.pos_enc[:x.shape[2]].cuda()
39
40
41	class SelfAttention(nn.Module):

nothing calls this directly

no outgoing calls

no test coverage detected