Searched refs:shuffles_inverse (Results 1 – 1 of 1) sorted by relevance
191 array<ptrdiff_t, 4> shuffles_inverse; in test_shuffle_unshuffle() local194 shuffles_inverse[shuffles[index]] = i; in test_shuffle_unshuffle()199 shuffle = tensor.shuffle(shuffles).shuffle(shuffles_inverse); in test_shuffle_unshuffle()