доступ к тензорным значениям внутри функции отображения набора данных в тензорном потоке - PullRequest
0 голосов
/ 08 марта 2020

Насколько я знаю, использование метода numpy () невозможно в функции карты в наборе данных tenorflow. Так как я могу получить значение тензора там? Я хочу преобразовать тензор в int, чтобы использовать его в качестве индекса словаря.

def process_path(file_path):
  print('file_path in process_path:', file_path)
  label = tf.strings.split(tf.strings.split(file_path, '/')[-1], '.')[-2]
  print('label in process_path:', label, type(label))
  label = tf.strings.to_number(label, tf.int64)

  labelint = labels['Male'][label]
  print(labelint)
  img = tf.io.read_file(file_path)
  img = decode_img(img)
  return img, label

Я хочу использовать приведенный выше код в функции карты, но получаю ошибку! Я ценю вашу помощь.

...