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])
