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 mnp_ravel(input_tensor): a = mnp.ravel(input_tensor) return a