Приведение тензора строки к списку строк - PullRequest
0 голосов
/ 12 апреля 2020

Как кто-то может привести следующий тензор потока строк:

<tf.Tensor: shape=(64,), dtype=string, numpy=
array([b'example string 1',
       b'example string 2',
       b'example string 3',

       ...

       b'example string 63',
       b'example string 64'],
      dtype=object)>

к списку строк:

[
    'example string 1',
    'example string 2',
    'example string 3',

       ...

    'example string 63',
    'example string 64'
]

, применяя векторизованный подход?

Спасибо заранее.

1 Ответ

1 голос
/ 12 апреля 2020

Просто используйте list(tensor.numpy()). Пример:

import tensorflow as tf

n_strings = 8
t = tf.convert_to_tensor(['example string ' + str(i) for i in range(n_strings)])
t
# <tf.Tensor: shape=(8,), dtype=string, numpy=
# array([b'example string 0', b'example string 1', b'example string 2',
#        b'example string 3', b'example string 4', b'example string 5',
#        b'example string 6', b'example string 7'], dtype=object)>
list(t.numpy().astype('str'))
# ['example string 0',
#  'example string 1',
#  'example string 2',
#  'example string 3',
#  'example string 4',
#  'example string 5',
#  'example string 6',
#  'example string 7']
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...