Function _transpose_dim_to_front

tensorflow/python/ops/parallel_for/pfor.py:2532–2539

(x, dim)

def _transpose_dim_to_front(x, dim):
  rank = array_ops.rank(x)
  return array_ops.transpose(
      x,
      perm=array_ops.concat([
          [dim],
          math_ops.range(0, dim),
          math_ops.range(dim + 1, rank)], axis=0))


@RegisterPForWithArgs("RandomUniform")