andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(nn/rnn): fixed wrong type assertion

Browse Source
This commit is contained in:
sugarme 2020-07-28 16:15:57 +10:00
parent 74480db4d6
commit a72ded34a3
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nn/rnn.go
View File

@ -267,8 +267,7 @@ func (g GRU) Seq(input ts.Tensor) (output ts.Tensor, state State) {
output, state = g.SeqInit(input, inState)
// Delete intermediate tensors in inState
inState.(LSTMState).Tensor1.MustDrop()
inState.(LSTMState).Tensor2.MustDrop()
inState.(GRUState).Tensor.MustDrop()
return output, state
}