Returns a bytes_list from a string / byte.
(value)
| 16 | |
| 17 | |
| 18 | def _bytes_feature(value): |
| 19 | """Returns a bytes_list from a string / byte.""" |
| 20 | if isinstance(value, type(tf.constant(0))): |
| 21 | value = value.numpy( |
| 22 | ) # BytesList won't unpack a string from an EagerTensor. |
| 23 | value = value if isinstance(value, list) else [value] |
| 24 | return tf.train.Feature(bytes_list=tf.train.BytesList(value=value)) |
| 25 | |
| 26 | |
| 27 | def _float_feature(value): |
no test coverage detected