@staticmethod
def forward(ctx, x, order):
    ctx.save_for_backward(order)
    return np.transpose(x, order)
@staticmethod
def forward(ctx, x, order):
    ctx.save_for_backward(order)
    return np.transpose(x, order)