def _test_runner(options): expr = create_expression.create_expression_from_prensor( prensor_test_util.create_simple_prensor()) new_root = map_prensor.map_sparse_tensor(expr, path.Path([]), [path.Path(["foo"])], lambda x: x * 2, True, tf.int32, "foo_doubled") leaf_node = expression_test_util.calculate_value_slowly( new_root.get_descendant_or_error(path.Path(["foo_doubled"])), options=options) self.evaluate(leaf_node.parent_index) self.evaluate(leaf_node.values)
def test_map_sparse_tensor(self): expr = create_expression.create_expression_from_prensor( prensor_test_util.create_simple_prensor()) new_root = map_prensor.map_sparse_tensor(expr, path.Path([]), [path.Path(["foo"])], lambda x: x * 2, False, tf.int32, "foo_doubled") leaf_node = expression_test_util.calculate_value_slowly( new_root.get_descendant_or_error(path.Path(["foo_doubled"]))) self.assertAllEqual(leaf_node.parent_index, [0, 1, 2]) self.assertAllEqual(leaf_node.values, [18, 16, 14])
def _test_runner(options): with self.session(use_gpu=False) as sess: expr = create_expression.create_expression_from_prensor( prensor_test_util.create_simple_prensor()) new_root = map_prensor.map_sparse_tensor( expr, path.Path([]), [path.Path(["foorepeated"])], lambda x: x * 2, False, tf.int32, "foorepeated_doubled") leaf_node = expression_test_util.calculate_value_slowly( new_root.get_descendant_or_error( path.Path(["foorepeated_doubled"])), options=options) sess.run([leaf_node.parent_index, leaf_node.values])
def test_map_sparse_tensor_one_output(self): for options in options_to_test: expr = create_expression.create_expression_from_prensor( _create_one_value_prensor()) new_root = map_prensor.map_sparse_tensor(expr, path.Path( []), [path.Path(["foo"])], lambda x: x * 2, False, tf.int32, "foo_doubled") leaf_node = expression_test_util.calculate_value_slowly( new_root.get_descendant_or_error(path.Path(["foo_doubled"])), options=options) self.assertAllEqual(leaf_node.parent_index, [1]) self.assertAllEqual(leaf_node.values, [16])
def test_map_sparse_tensor_is_repeated(self): for options in options_to_test: expr = create_expression.create_expression_from_prensor( prensor_test_util.create_simple_prensor()) new_root = map_prensor.map_sparse_tensor( expr, path.Path([]), [path.Path(["foorepeated"])], lambda x: x * 2, True, tf.int32, "foorepeated_doubled") leaf_node = expression_test_util.calculate_value_slowly( new_root.get_descendant_or_error( path.Path(["foorepeated_doubled"])), options=options) self.assertAllEqual(leaf_node.parent_index, [0, 1, 1, 2]) self.assertAllEqual(leaf_node.values, [18, 16, 14, 12])