如何在无限生成器中重置变量？

def infinite_generator():
    count = 0
    while True:
        reset = yield count
        if reset:
            count = 0
        else:
            count += 1

# 创建无限生成器
generator = infinite_generator()

# 生成值并打印
print(next(generator))  # 输出: 0
print(next(generator))  # 输出: 1
print(next(generator))  # 输出: 2

# 重置变量并生成值
print(generator.send(True))  # 输出: 0
print(next(generator))  # 输出: 1
print(next(generator))  # 输出: 2