Yunzhe Hao

icon company CASIA icon location Beijing China icon location PhD student @ CASIA; Main interests include Speech Separation, Auditory Attention, Spiking Neural Network

Results 1 comments of Yunzhe Hao

How to write cuda code of the multilayer units

There are two classes about lstm in pytorch, LSTMCell and LSTM. The former is a single layer and can just receive one step input. The latter can stack several layers,...