Python pad_tensor Examples

Example #1

    def graph_fn():
      t1 = tf.constant([1], dtype=tf.int32)
      pad_t1 = shape_utils.pad_tensor(t1, tf.constant(2))
      t2 = tf.constant([[0.1, 0.2]], dtype=tf.float32)
      pad_t2 = shape_utils.pad_tensor(t2, tf.constant(2))

      return pad_t1, pad_t2

Example #2

    def test_pad_tensor_using_tensor_input(self):
        t1 = tf.constant([1], dtype=tf.int32)
        pad_t1 = shape_utils.pad_tensor(t1, tf.constant(2))
        t2 = tf.constant([[0.1, 0.2]], dtype=tf.float32)
        pad_t2 = shape_utils.pad_tensor(t2, tf.constant(2))

        with self.test_session() as sess:
            pad_t1_result, pad_t2_result = sess.run([pad_t1, pad_t2])
            self.assertAllEqual([1, 0], pad_t1_result)
            self.assertAllClose([[0.1, 0.2], [0, 0]], pad_t2_result)

Example #3

File: shape_utils_test.py Project: ALISCIFP/models

  def test_pad_tensor_using_tensor_input(self):
    t1 = tf.constant([1], dtype=tf.int32)
    pad_t1 = shape_utils.pad_tensor(t1, tf.constant(2))
    t2 = tf.constant([[0.1, 0.2]], dtype=tf.float32)
    pad_t2 = shape_utils.pad_tensor(t2, tf.constant(2))

    with self.test_session() as sess:
      pad_t1_result, pad_t2_result = sess.run([pad_t1, pad_t2])
      self.assertAllEqual([1, 0], pad_t1_result)
      self.assertAllClose([[0.1, 0.2], [0, 0]], pad_t2_result)