Method forward

ioblocks.py:91–92

(self, x)

        self.output = nn.Linear(dim, vocab_size, bias=False)

    def forward(self, x):
        return self.output(x)


# helper functions