def _test_property(self, fn):
for constructor in _iter_constructors():
num_nested_tensor = 3
nested_tensor_lists = [utils.gen_nested_list(i, i, 3)
for i in range(1, num_nested_tensor)]
first_tensors = [utils.get_first_tensor(
ntl) for ntl in nested_tensor_lists]
nested_tensors = [constructor(ntl) for ntl in nested_tensor_lists]
for nested_tensor, first_tensor in zip(nested_tensors, first_tensors):
self.assertEqual(fn(nested_tensor), fn(first_tensor))
def _test_property(self, fn):
for constructor in _iter_constructors():
# TODO: Used to be 3. Currently only supporting nested dim 1.
num_nested_tensor = 1
nested_tensor_lists = [utils.gen_nested_list(i, i, 3)
for i in range(1, num_nested_tensor)]
first_tensors = [utils.get_first_tensor(
ntl) for ntl in nested_tensor_lists]
nested_tensors = [constructor(ntl) for ntl in nested_tensor_lists]
for nested_tensor, first_tensor in zip(nested_tensors, first_tensors):
self.assertEqual(fn(nested_tensor), fn(first_tensor))