Использование ops из Tensorflow.contrb в Windows через API Java

Question

Я хотел бы загрузить op из tf.contrib, в частности "_beam_search_ops". Я использую Tensorflow 1.6 от Maven.

После ответа на Проблема при загрузке / обслуживании модели tenorflow в Java с использованием оценщиков , я могу это сделать в Ubuntu 17.10.

Код выглядит так:

TensorFlow.loadLibrary("_beam_search_ops.so");

Затем я пытаюсь сделать то же самое в Windows 10 с кодом

TensorFlow.loadLibrary("_beam_search_ops.dll");

И это не с

java.lang.UnsatisfiedLinkError: D:\Users\ALEXAN~1\AppData\Local\Temp\deep_api_plugin\_beam_search_ops.dll not found
    at org.tensorflow.TensorFlow.loadLibrary(TensorFlow.java:47)

Так в чем же причина этого?

Answer 1

Пока я писал и исследовал вопрос, я наткнулся на решение.

Чтобы решить эту проблему, файл _pywrap_tensorflow_internal.pyd следует скопировать в папку, в которой находится _beam_search_ops.dll. Я понимаю, что это тоже какой-то dll, но не понимаю, почему это нужно для Windows, а не для Ubuntu. Если кто-то знает, пожалуйста, просветите меня.

Также я понимаю, что такого рода дополнительная информация должна быть предоставлена ​​в комментарии к первоначальному ответу, но, к сожалению, у меня недостаточно репутации, чтобы комментировать, и я подумал, что этот ответ может кому-то помочь.

Редактировать: я был слишком быстр, чтобы праздновать. С этим ответом библиотека загружается без исключения, но модель, похоже, не может ее использовать, в случае ошибки

Exception in thread "main" org.tensorflow.TensorFlowException: Op type not registered 'GatherTree' in binary running on DESKTOP-EB6GIDR. Make sure the Op and Kernel are registered in the binary running in this process.
    at org.tensorflow.SavedModelBundle.load(Native Method)
    at org.tensorflow.SavedModelBundle.load(SavedModelBundle.java:39)

Поиск ответа должен продолжаться.