函数返回列表

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])]

由此可知输出的是一个列表。