Method cast

tensorflow/python/ops/init_ops.py:1099–1101

Return p or (1-p).

(i, p)

      return math_ops.matmul(math_ops.matmul(p1, p2), p3)

    def cast(i, p):
      """Return p or (1-p)."""
      return i * p + (1 - i) * (eye - p)

    for i in [0, 1]:
      for j in [0, 1]: