(self, x, scale=1)
| 50 | self.register_buffer('pe', pe) |
| 51 | |
| 52 | def forward(self, x, scale=1): |
| 53 | return self.pe[:, scale:x.size(1)*scale+1:scale] |
| 54 | |
| 55 | class TokenEmbedding(nn.Module): |
| 56 | def __init__(self, c_in, d_model): |
nothing calls this directly
no outgoing calls
no test coverage detected