def __init__(self, strategy1, strategy2):
    super(TransposeNet, self).__init__()
    self.matmul = P.MatMul().set_strategy(((8, 1), (1, 1)))
    self.matmul_weight = Parameter(Tensor(np.ones([128, 256]), dtype=ms.float32), name='weight')
    self.transpose1 = P.Transpose().set_strategy(strategy1)
    self.transpose2 = P.Transpose().set_strategy(strategy2)
def __init__(self, strategy1, strategy2):
    super(TransposeNet, self).__init__()
    self.matmul = P.MatMul().set_strategy(((8, 1), (1, 1)))
    self.matmul_weight = Parameter(Tensor(np.ones([128, 256]), dtype=ms.float32), name='weight')
    self.transpose1 = P.Transpose().set_strategy(strategy1)
    self.transpose2 = P.Transpose().set_strategy(strategy2)