Using ParallelInference to get a specific layer

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.

@mdavis95 file an issue and I’ll take a look at adding this. Thanks!

Thanks, ticket is here: Support using ParallelInference to get the output of an internal layer · Issue #9841 · deeplearning4j/deeplearning4j · GitHub. Let me know if I need to add more information to the ticket.