本文共 459 字,大约阅读时间需要 1 分钟。
首先,先直观上理解一下两种返回方式的区别:
然后,概括的理解一下:
yield 能和 for 循环共同使用最后,两种返回值,在 for 循环上的区别:
import torchdef fun(): for i in range(3): a = torch.tensor([[1,2]]) b = torch.tensor([[3,4,5]]) return a,bi=0for a in fun(): i += 1 print('step:',i) print(a)
import torchdef fun(): for i in range(3): a = torch.tensor([[1,2]]) b = torch.tensor([[3,4,5]]) yield a,bi=0for a in fun(): i += 1 print('step:',i) print(a)
转载地址:http://tpqqf.baihongyu.com/