diff options
Diffstat (limited to 'tensorflow/go/shape.go')
| -rw-r--r-- | tensorflow/go/shape.go | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/go/shape.go b/tensorflow/go/shape.go index 8d000cb9de..114ab5decb 100644 --- a/tensorflow/go/shape.go +++ b/tensorflow/go/shape.go @@ -59,7 +59,7 @@ func (s Shape) NumDimensions() int { // // REQUIRES: 0 <= dim < s.NumDimensions() func (s Shape) Size(dim int) int64 { - if dim < 0 || dim >= s.NumDimensions() { + if dim < 0 || dim > s.NumDimensions() { return -1 } return s.dims[dim] |