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.