def construct(self, x):
    x = self.matmul(x, self.matmul_weight)
    x = self.transpose1(x, (1, 0))
    x = self.transpose2(x, (1, 0))
    return x
def construct(self, x):
    x = self.matmul(x, self.matmul_weight)
    x = self.transpose1(x, (1, 0))
    x = self.transpose2(x, (1, 0))
    return x