Function TransposeOperand

tensorflow/core/kernels/einsum_op.cc:323–337

// transposing is not necessary.
template <typename Device, typename T>
Status TransposeOperand(OpKernelContext* ctx, const Tensor& input,
                        const std::vector<int>& permutation, Tensor* output) {
  if (!ShouldTranspose(input.shape(), permutation)) {
    return CopyFrom(input, input.shape(), output);
  }
  TensorShape transposed_shape;
  for (int i = 0; i < input.dims(); ++i) {
    transposed_shape.AddDim(input.dim_size(permutation[i]));
  }
  TF_RETURN_IF_ERROR(
      ctx->allocate_temp(DataTypeToEnum<T>::value, transposed_shape, output));
  const Device& device = ctx->eigen_device<Device>();
  TF_RETURN_IF_ERROR(DoTranspose(device, input, permutation, output));
  return Status::OK();
}

// If there are repeated labels in either the input or output, then this strides
// the input (e.g. iii->i) or inflates it (e.g. i->iii), respectively.