Pytorch 记录每次循环输出的结果

toolkit: pytorch
需求:对每次循环生成的结果,stack,在循环结束后输出。

out = torch.tensor([])

for i in range(100):
	out = torch.stack(
		(
			*out, 
			func(xxxx)
		),
		dim = 0 #default
		)


你可能感兴趣的:(pytorch,代码技巧,pytorch,人工智能,python)