def mnp_transpose(input_tensor):
a = mnp.transpose(input_tensor, (0, 2, 1))
b = mnp.transpose(input_tensor, [2, 1, 0])
c = mnp.transpose(input_tensor, (1, 0, 2))
d = mnp.transpose(input_tensor)
return a, b, c, d
def construct(self, x1, x2, x3):
x1 = mnp.transpose(x1, [0, 2, 1])
x2 = x2.transpose(0, 2, 1)
x = mnp.concatenate((x1, x2, x3), -1)
x = mnp.ravel(x)
return x
def transpose(a, axes=None):
return mnp.transpose(a, axes)
