Fixed the tensor shuffling test
5 files changed