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
相关文章
相关标签/搜索