I have the following input with the shape (1,1,2,2,2,2):
x = array([[[[[[0., 0.], [0., 1.]], [[0., 0.], [1., 0.]]],