Method Len

tensor/shape.go:58–67

Len returns the total length of elements in the tensor (i.e., the product of the shape sizes)

()

// Len returns the total length of elements in the tensor
// (i.e., the product of the shape sizes)
func (sh *Shape) Len() int {
	if len(sh.Sizes) == 0 {
		return 0
	}
	o := int(1)
	for _, v := range sh.Sizes {
		o *= v
	}
	return int(o)
}

// NumDims returns the total number of dimensions.
func (sh *Shape) NumDims() int { return len(sh.Sizes) }