Method __init__

exir/tests/models.py:196–201

(self, transposed: bool = False)

class BatchMatrixMultiplication(nn.Module):
    def __init__(self, transposed: bool = False) -> None:
        super().__init__()

        # Whether the last 2 dims (-1, -2) of the input has already been
        # transposed. If yes, transpose it back before feeding to torch.bmm
        self.transposed: bool = transposed

    def forward(self, x: Tensor, y: Tensor) -> Tensor:
        if self.transposed: