permutedims2! fix