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

fixed nn.Seq Forward nil pointer if layers length = 1

Browse Source
This commit is contained in:
sugarme 2021-05-15 17:50:08 +10:00
parent 9135395bce
commit 720beffa62
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nn/sequential.go
View File

@ -81,6 +81,10 @@ func (s *Sequential) Forward(xs *ts.Tensor) (retVal *ts.Tensor) {
return xs.MustShallowClone()
}
if len(s.layers) == 1 {
return s.layers[0].Forward(xs)
}
// forward sequentially
outs := make([]ts.Tensor, len(s.layers))
for i := 0; i < len(s.layers); i++ {