Method forward

vtp/models/layers/ffn.py:12–13

(self, x: Tensor)

    """Mixin for forward_list operation on list of tensors."""

    def forward(self, x: Tensor):
        raise NotImplementedError

    def forward_list(self, x_list: List[Tensor]) -> List[Tensor]:
        x_flat, shapes, num_tokens = cat_keep_shapes(x_list)