Function NewIterator

niterator/onearrpremul/iterator.go:24–36

(ap *shape.AP)

}

func NewIterator(ap *shape.AP) *Iterator {
	track := make([]Index, len(ap.Shape))
	for i := range ap.Shape {
		track[i].Shape = uint32(ap.Shape[i])
		track[i].Stride = uint32(ap.Stride[i])
		track[i].Premul = uint32((ap.Shape[i] - 1) * ap.Stride[i])
	}

	return &Iterator{
		AP:    ap,
		Track: track,
	}
}

func (it *Iterator) IsDone() bool {
	return it.Done