permute_copy_i32

void permute_copy_i32(const int * in, int * out, const int & in_shape, const int & perm)

Int32 counterpart to permute_copy_f32.

Used for argmax / index tensors that need to be re-laid out as part of a larger op (e.g. transposed indexing).

inconst int32_t*

Source buffer laid out densely (C-order) according to in_shape.

outint32_t*

Pre-allocated destination buffer.

in_shapeconst std::vector<int64_t>&

Shape of the source tensor.

permconst std::vector<int>&

Axis permutation; see permute_copy_f32.