Method Size

tensorflow/go/shape.go:61–66

Size returns the size of the dim-th dimension of the shape, or -1 if it is unknown. REQUIRES: 0 <= dim < s.NumDimensions()

(dim int)

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

// IsFullySpecified returns true iff the size of all the dimensions of s are
// known.