Function ShouldTranspose

tensorflow/core/kernels/einsum_op.cc:311–318

Returns whether transposing would be a no-op; whether input has rank < 2 or the permutation is the identity permutation.

// Returns whether transposing would be a no-op; whether input has rank < 2 or
// the permutation is the identity permutation.
bool ShouldTranspose(const TensorShape& input_shape,
                     const std::vector<int>& permutation) {
  if (input_shape.dims() < 2) return false;
  for (int i = 0; i < permutation.size(); ++i) {
    if (permutation[i] != i) return true;
  }
  return false;
}

// Transpose the input given a permutation. Returns a reference to the input if
// transposing is not necessary.