python 中 return VS yield

首先，先直观上理解一下两种返回方式的区别：python中yield的用法详解——最简单，最清晰的解释

然后，概括的理解一下：python-复盘-yield & return区别 yield 能和 for 循环共同使用

最后，两种返回值，在 for 循环上的区别：

import torch def fun(): for i in range(3): a = torch.tensor([[1,2]]) b = torch.tensor([[3,4,5]]) return a,b i=0 for a in fun(): i += 1 print('step:',i) print(a)

import torch def fun(): for i in range(3): a = torch.tensor([[1,2]]) b = torch.tensor([[3,4,5]]) yield a,b i=0 for a in fun(): i += 1 print('step:',i) print(a)