import tensorflow as tf


  Returns
  -------
  A tensor.
  """
  x = (0.2 * x) + 0.5
  zero = _to_tensor(0., x.dtype.base_dtype)
  one = _to_tensor(1., x.dtype.base_dtype)
  x = tf.clip_by_value(x, zero, one)
  return x


def sqrt(x):
  """Element-wise square root.

  Parameters
