DeepLearning4j的StackVertex實現參數共享

設input1和input2是shape爲(M,I)的張量，權重W是shape爲(I,O)的張量, 設對應的兩個輸出值爲output1和output2，其shape是(M,O)，設損失函數對輸出值的導數爲epsilon1和epsilon2，損失函數對線性部分的導數爲delta1和delta2，shape也都是(M,O)： Nd4j.gemm(input, delta, weightGrad, tr

>>阅读原文<<