def test_is_tensor_array(self): self.assertTrue(tensors.is_tensor_array(self._simple_tensor_array())) self.assertFalse(tensors.is_tensor_array(self._simple_tensor_list())) self.assertFalse(tensors.is_tensor_array(constant_op.constant(1))) self.assertFalse( tensors.is_tensor_array(self._simple_list_of_tensors())) self.assertFalse(tensors.is_tensor_array(None))
def len_(s): if tensors.is_tensor_array(s): return _tf_tensor_array_len(s) elif tensors.is_tensor_list(s): return _tf_tensor_list_len(s) elif tensor_util.is_tensor(s): return _tf_tensor_len(s) return _py_len(s)
def len_(s): if tensors.is_tensor_array(s): return _tf_tensor_array_len(s) elif tensors.is_tensor_list(s): return _tf_tensor_list_len(s) elif tensor_util.is_tf_type(s): return _tf_tensor_len(s) if isinstance(s, dataset_ops.DatasetV2): return _tf_dataset_len(s) return _py_len(s)
def test_is_tensor_array(self): self.assertTrue(tensors.is_tensor_array(self._simple_tensor_array())) self.assertFalse(tensors.is_tensor_array(self._simple_tensor_list())) self.assertFalse(tensors.is_tensor_array(constant_op.constant(1))) self.assertFalse(tensors.is_tensor_array(self._simple_list_of_tensors())) self.assertFalse(tensors.is_tensor_array(None))