Как использовать tf.withcontrodependences в API Tensorflow Java? ⇐ JAVA

[Anonymous]

Я хочу поместить два тензора в карту, а затем получить один тензор с карты. Чтобы обеспечить выполнение tf.tensormapinsert , я использую tf.withcontrodependences - но он не работает.
Вот мой код:

Код: Выделить всё

Operand map = tf.emptyTensorMap();
Operand out = tf.withControlDependencies(
tf.tensorMapInsert(map, tf.constant("hidden"), hNext),
tf.tensorMapInsert(map, tf.constant("cell"), cNext)
).identity(map);
out = tf.tensorMapLookup(out, tf.constant("hidden"), TFloat32.class);

Когда я извлекаю , я получил ошибку: попытка поиска несуществующего ключа. Не удалось найти ключ «скрытый».
Полный стек:

Код: Выделить всё

2025-05-25 02:15:07.809978: W tensorflow/core/framework/local_rendezvous.cc:404] Local rendezvous is aborting with status: INVALID_ARGUMENT: Trying to lookup non-existent key. Could not find key "hidden".
[[{{node TensorMapLookup}}]]

org.tensorflow.exceptions.TFInvalidArgumentException: Trying to lookup non-existent key. Could not find key "hidden".
[[{{node TensorMapLookup}}]]

at org.tensorflow.internal.c_api.AbstractTF_Status.throwExceptionIfNotOK(AbstractTF_Status.java:76)
at org.tensorflow.Session.run(Session.java:826)
at org.tensorflow.Session$Runner.runHelper(Session.java:549)
at org.tensorflow.Session$Runner.run(Session.java:476)
at org.apache.wayang.tensorflow.model.TensorflowOperatorTest.testConvLSTM2D(TensorflowOperatorTest.java:134)
at java.base/jdk.internal.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at java.base/jdk.internal.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:77)
at java.base/jdk.internal.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
at java.base/java.lang.reflect.Method.invoke(Method.java:569)
at org.junit.runners.model.FrameworkMethod$1.runReflectiveCall(FrameworkMethod.java:59)
at org.junit.internal.runners.model.ReflectiveCallable.run(ReflectiveCallable.java:12)
at org.junit.runners.model.FrameworkMethod.invokeExplosively(FrameworkMethod.java:56)
at org.junit.internal.runners.statements.InvokeMethod.evaluate(InvokeMethod.java:17)
at org.junit.runners.ParentRunner$3.evaluate(ParentRunner.java:306)
at org.junit.runners.BlockJUnit4ClassRunner$1.evaluate(BlockJUnit4ClassRunner.java:100)
at org.junit.runners.ParentRunner.runLeaf(ParentRunner.java:366)
at org.junit.runners.BlockJUnit4ClassRunner.runChild(BlockJUnit4ClassRunner.java:103)
at org.junit.runners.BlockJUnit4ClassRunner.runChild(BlockJUnit4ClassRunner.java:63)
at org.junit.runners.ParentRunner$4.run(ParentRunner.java:331)
at org.junit.runners.ParentRunner$1.schedule(ParentRunner.java:79)
at org.junit.runners.ParentRunner.runChildren(ParentRunner.java:329)
at org.junit.runners.ParentRunner.access$100(ParentRunner.java:66)
at org.junit.runners.ParentRunner$2.evaluate(ParentRunner.java:293)
at org.junit.runners.ParentRunner$3.evaluate(ParentRunner.java:306)
at org.junit.runners.ParentRunner.run(ParentRunner.java:413)
at org.junit.runner.JUnitCore.run(JUnitCore.java:137)
at com.intellij.junit4.JUnit4IdeaTestRunner.startRunnerWithArgs(JUnit4IdeaTestRunner.java:69)
at com.intellij.rt.junit.IdeaTestRunner$Repeater$1.execute(IdeaTestRunner.java:38)
at com.intellij.rt.execution.junit.TestsRepeater.repeat(TestsRepeater.java:11)
at com.intellij.rt.junit.IdeaTestRunner$Repeater.startRunnerWithArgs(IdeaTestRunner.java:35)
at com.intellij.rt.junit.JUnitStarter.prepareStreamsAndStart(JUnitStarter.java:232)
at com.intellij.rt.junit.JUnitStarter.main(JUnitStarter.java:55)

Так как использовать tf.withcontroldependences в tensorflow java api?

Подробнее здесь: https://stackoverflow.com/questions/796 ... w-java-api