toolkit: pytorch
需求:对每次循环生成的结果,stack,在循环结束后输出。
out = torch.tensor([])
for i in range(100):
out = torch.stack(
(
*out,
func(xxxx)
),
dim = 0 #default
)
toolkit: pytorch
需求:对每次循环生成的结果,stack,在循环结束后输出。
out = torch.tensor([])
for i in range(100):
out = torch.stack(
(
*out,
func(xxxx)
),
dim = 0 #default
)