python中的yield关键字是个什么鬼?它有什么用处?_generator函数的yield关键字的作用是_向东的笔记本的博客-CSDN博客
import torch
def foo(): # 生成器
for i in range(10):
yield i,"hi" # 迭代器
g = foo()
# 法1:一个个获取
print(next(g)) #(0, 'hi')
print(next(g)) #(1, 'hi')
# 法2: 用for自动全部获取
for x in g: # for 会调用 iter()函数获取生成器,然后调用next()
print(x)