LSTM矩阵运算三图流
2019-08-29 本文已影响0人
tianchen627
1
2
3
注意图中h的hidden size和x的embedding size都为h，事实上并不一定需要相同。考虑单个输入输出的i，f，o，g均为标量；若输入的是batch，则图中x和h不再是h x 1向量而是h x batchsize矩阵，输出的i，f，o，g为batchsize x 1向量。
1
2
3
1
2
3
1
2
3
注意图中h的hidden size和x的embedding size都为h，事实上并不一定需要相同。考虑单个输入输出的i，f，o，g均为标量；若输入的是batch，则图中x和h不再是h x 1向量而是h x batchsize矩阵，输出的i，f，o，g为batchsize x 1向量。
