deeplearning4j / deeplearning4j

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

Home Page:http://deeplearning4j.konduit.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to extract the last time steps from two or more LSTM networks and add several density layers to form a computational graph?

cqiaoYc opened this issue · comments

I want to use 2-3 LSTM networks as feature selectors, extract the last time steps of LSTM networks as the input layer, and then add several density layers to build a computation graph. How can I use a computation graph to build it?

@cqiaoYc please post on the forums. You know this by now. Thanks!