def a():
w=torch.randn((5,1))
b=torch.randn(1)
return [w,b]
return 函数返回的是一个列表。
代码展示:
def a():
w=torch.randn((5,1))
b=torch.randn(1)
return [w,b]
print(a())
[tensor([[-1.2416],
[-0.0185],
[-1.2681],
[ 0.8797],
[-0.8969]]), tensor([-1.3992])]
由此可知输出的是一个列表。