Found a problem in benchmarks.py

Hi, I found an issue in benchmark.py at line 92 and 116. There is a np.concatenate operation that expands the dimension of (B, trans , J , 3) at axis 2. The original number of joints is 22. After that operation, the joints number is 23.