in tensorflow_gnn/graph/graph_tensor_pprint.py [0:0]
def _get_tensor_data(listorarray: Any) -> Any:
"""Convert a tensor to plain-old data types."""
if isinstance(listorarray, (int, float, str)):
return listorarray
elif isinstance(listorarray, bytes):
try:
return listorarray.decode('utf8')
except UnicodeDecodeError:
# Note: This may be useful for images and other non-text binary data.
return listorarray
elif isinstance(listorarray, list):
return [_get_tensor_data(elem) for elem in listorarray]
elif isinstance(listorarray, numpy.ndarray):
return [_get_tensor_data(elem) for elem in listorarray.tolist()]
else:
raise TypeError(f'Unsupported type: {listorarray}')