Update TensorIterator.h