experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix shape.go's Size function.

Browse Source 
Fixes #10741.
This commit is contained in:
Andrew Selle 2017-06-15 14:56:36 -07:00 committed by Martin Wicke
parent 319c847d94
commit 76a0a15cb9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/go/shape.go
View File

@ -59,7 +59,7 @@ func (s Shape) NumDimensions() int {
// //
// REQUIRES: 0 <= dim < s.NumDimensions() // REQUIRES: 0 <= dim < s.NumDimensions()
func (s Shape) Size(dim int) int64 { func (s Shape) Size(dim int) int64 {
if dim < 0 || dim > s.NumDimensions() { if dim < 0 || dim >= s.NumDimensions() {
return -1 return -1
} }
return s.dims[dim] return s.dims[dim]