Python Reshape.forward方法代码示例

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
  def testScalarReshape(self):
    """Test reshaping to and from a scalar shape ()."""

    expected_x = np.random.randn(4, 3, 1)
    expected_y = np.reshape(expected_x, [4, 3])

    expected_x_scalar = np.random.randn(1,)
    expected_y_scalar = expected_x_scalar[0]

    shape_in, shape_out, feed_dict = self.build_shapes([], [1,])
    with self.cached_session() as sess:
      bijector = Reshape(
          event_shape_out=shape_in,
          event_shape_in=shape_out, validate_args=True)
      (x_,
       y_,
       x_scalar_,
       y_scalar_
      ) = sess.run((
          bijector.inverse(expected_y),
          bijector.forward(expected_x),
          bijector.inverse(expected_y_scalar),
          bijector.forward(expected_x_scalar),
      ), feed_dict=feed_dict)
      self.assertAllClose(expected_y, y_, rtol=1e-6, atol=0)
      self.assertAllClose(expected_x, x_, rtol=1e-6, atol=0)
      self.assertAllClose(expected_y_scalar, y_scalar_, rtol=1e-6, atol=0)
      self.assertAllClose(expected_x_scalar, x_scalar_, rtol=1e-6, atol=0)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
  def testBijector(self):
    """Do a basic sanity check of forward, inverse, jacobian."""
    expected_x = np.random.randn(4, 3, 2)
    expected_y = np.reshape(expected_x, [4, 6])

    with self.cached_session() as sess:
      shape_in, shape_out, feed_dict = self.build_shapes([3, 2], [6,])
      bijector = Reshape(
          event_shape_out=shape_out,
          event_shape_in=shape_in,
          validate_args=True)
      (x_,
       y_,
       fldj_,
       ildj_) = sess.run((
           bijector.inverse(expected_y),
           bijector.forward(expected_x),
           bijector.forward_log_det_jacobian(expected_x, event_ndims=2),
           bijector.inverse_log_det_jacobian(expected_y, event_ndims=2),
       ), feed_dict=feed_dict)
      self.assertEqual("reshape", bijector.name)
      self.assertAllClose(expected_y, y_, rtol=1e-6, atol=0)
      self.assertAllClose(expected_x, x_, rtol=1e-6, atol=0)
      self.assertAllClose(0., fldj_, rtol=1e-6, atol=0)
      self.assertAllClose(0., ildj_, rtol=1e-6, atol=0)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
  def testValidButNonMatchingInputPartiallySpecifiedOpError(self):
    x = np.random.randn(4, 3, 2)

    with self.cached_session() as sess:
      shape_in, shape_out, feed_dict = self.build_shapes([2, -1], [1, 6, 1,])
      bijector = Reshape(
          event_shape_out=shape_out,
          event_shape_in=shape_in,
          validate_args=True)

      with self.assertRaisesError(
          "Input `event_shape` does not match `event_shape_in`."):
        sess.run(bijector.forward(x),
                 feed_dict=feed_dict)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
 def testDefaultVectorShape(self):
   expected_x = np.random.randn(4, 4)
   expected_y = np.reshape(expected_x, [4, 2, 2])
   with self.cached_session() as sess:
     _, shape_out, feed_dict = self.build_shapes([-1,], [-1, 2])
     bijector = Reshape(shape_out,
                        validate_args=True)
     (x_,
      y_,
     ) = sess.run((
         bijector.inverse(expected_y),
         bijector.forward(expected_x),
     ), feed_dict=feed_dict)
     self.assertAllClose(expected_y, y_, rtol=1e-6, atol=0)
     self.assertAllClose(expected_x, x_, rtol=1e-6, atol=0)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
  def _testInputOutputMismatchOpError(self, expected_error_message):
    x1 = np.random.randn(4, 2, 3)
    x2 = np.random.randn(4, 1, 1, 5)

    with self.cached_session() as sess:
      shape_in, shape_out, fd_mismatched = self.build_shapes([2, 3],
                                                             [1, 1, 5])
      bijector = Reshape(
          event_shape_out=shape_out,
          event_shape_in=shape_in,
          validate_args=True)

      with self.assertRaisesError(expected_error_message):
        sess.run(bijector.forward(x1), feed_dict=fd_mismatched)
      with self.assertRaisesError(expected_error_message):
        sess.run(bijector.inverse(x2), feed_dict=fd_mismatched)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
 def testBothShapesPartiallySpecified(self):
   expected_x = np.random.randn(4, 2, 3)
   expected_y = np.reshape(expected_x, [4, 3, 2])
   with self.cached_session() as sess:
     shape_in, shape_out, feed_dict = self.build_shapes([-1, 3], [-1, 2])
     bijector = Reshape(
         event_shape_out=shape_out,
         event_shape_in=shape_in,
         validate_args=True)
     (x_,
      y_,
     ) = sess.run((
         bijector.inverse(expected_y),
         bijector.forward(expected_x),
     ), feed_dict=feed_dict)
     self.assertAllClose(expected_y, y_, rtol=1e-6, atol=0)
     self.assertAllClose(expected_x, x_, rtol=1e-6, atol=0)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
  def testValidButNonMatchingInputOpError(self):
    x = np.random.randn(4, 3, 2)

    with self.cached_session() as sess:
      shape_in, shape_out, feed_dict = self.build_shapes([2, 3], [1, 6, 1,])
      bijector = Reshape(
          event_shape_out=shape_out,
          event_shape_in=shape_in,
          validate_args=True)

      # Here we pass in a tensor (x) whose shape is compatible with
      # the output shape, so tf.reshape will throw no error, but
      # doesn't match the expected input shape.
      with self.assertRaisesError(
          "Input `event_shape` does not match `event_shape_in`."):
        sess.run(bijector.forward(x),
                 feed_dict=feed_dict)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
  def testOneShapePartiallySpecified(self):
    expected_x = np.random.randn(4, 6)
    expected_y = np.reshape(expected_x, [4, 2, 3])

    with self.test_session() as sess:
      # one of input/output shapes is partially specified
      shape_in, shape_out, feed_dict = self.build_shapes([-1,], [2, 3])
      bijector = Reshape(
          event_shape_out=shape_out,
          event_shape_in=shape_in,
          validate_args=True)
      (x_,
       y_,
      ) = sess.run((
          bijector.inverse(expected_y),
          bijector.forward(expected_x),
      ), feed_dict=feed_dict)
      self.assertAllClose(expected_y, y_, rtol=1e-6, atol=0)
      self.assertAllClose(expected_x, x_, rtol=1e-6, atol=0)

# 需要导入模块: from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape [as 别名]
# 或者: from tensorflow.contrib.distributions.python.ops.bijectors.reshape.Reshape import forward [as 别名]
  def testInputOutputMismatchOpError(self):
    x1 = np.random.randn(4, 2, 3)
    x2 = np.random.randn(4, 1, 1, 5)

    with self.test_session() as sess:
      shape_in, shape_out, fd_mismatched = self.build_shapes([2, 3],
                                                             [1, 1, 5])
      bijector = Reshape(
          event_shape_out=shape_out,
          event_shape_in=shape_in,
          validate_args=True)

      # test that *all* methods check basic assertions
      with self.assertRaisesError(
          "Input to reshape is a tensor with"):
        sess.run(bijector.forward(x1), feed_dict=fd_mismatched)
      with self.assertRaisesError(
          "Input to reshape is a tensor with"):
        sess.run(bijector.inverse(x2), feed_dict=fd_mismatched)