I am currently using
public INDArray[] output(List<String> layers, boolean train, INDArray[] features, INDArray[] featureMasks)
to get the output of an embedding layer. When I tried to switch to using ParallelInference
I noticed there is no method with that signature. What is the recommended way of using ParallelInference
to get the output of an internal layer.