from torch import pi initial_conditions = [ [1/6 * pi, 0.0, 0.0, 0.0], [-1/6 * pi, 0.0, 0.0, 0.0], [2/3 * pi, 0.0, 0.0, 0.0], [-2/3 * pi, 0.0, 0.0, 0.0], [0.0, 1/3 * pi, 0.0, 0.0], [0.0, -1/3 * pi, 0.0, 0.0], [0.0, 2 * pi, 0.0, 0.0], [0.0, -2 * pi, 0.0, 0.0], [0.0, 0.0, 0.0, 2 * pi], [0.0, 0.0, 0.0, -2 * pi], [0.0, 0.0, 0.0, 1/2 * pi], [0.0, 0.0, 0.0, -1/2 * pi], [0.0, 0.0, 0.0, 1/3 * pi], [0.0, 0.0, 0.0, -1/3 * pi], [1/4 * pi, 1 * pi, 0.0, 0.0], [-1/4 * pi, -1 * pi, 0.0, 0.0], [1/2 * pi, -1 * pi, 0.0, 1/3 * pi], [-1/2 * pi, 1 * pi, 0.0, -1/3 * pi], [1/4 * pi, 1 * pi, 0.0, 2 * pi], [-1/4 * pi, -1 * pi, 0.0, 2 * pi], [1/2 * pi, -1 * pi, 0.0, 4 * pi], [-1/2 * pi, 1 * pi, 0.0, -4 * pi], ]