Python transpose_batch_time Examples

Example #1

File: tensor_utils_test.py Project: yyht/language

    def test_transpose_batch_time_rank_unknown(self):
        with self.test_session(graph=tf.Graph()):
            tensor = tf.placeholder(tf.float32, shape=None)

            with self.assertRaisesRegexp(ValueError,
                                         "Tensor with unknown rank"):
                tensor_utils.transpose_batch_time(tensor)

Example #2

File: tensor_utils_test.py Project: yyht/language

    def test_transpose_batch_time_rank_1(self):
        with self.test_session(graph=tf.Graph()):
            tensor = tf.constant([1, 2, 3])

            transposed = tensor_utils.transpose_batch_time(tensor)

            self.assertAllEqual(transposed.eval(), [1, 2, 3])

Example #3

File: tensor_utils_test.py Project: yyht/language

    def test_transpose_batch_time_rank_2(self):
        with self.test_session(graph=tf.Graph()):
            tensor = tf.constant([[1, 2], [3, 4]])  # pyformat: disable

            transposed = tensor_utils.transpose_batch_time(tensor)

            self.assertAllEqual(transposed.eval(),
                                [[1, 3], [2, 4]])  # pyformat: disable