yes, it's equivalent, but more memory efficient (memory ~ sum(shape), not product(shape)). I'll add a note