📜  torch.nn.Linear(in_features, out_features,bias=True) 描述 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:10.071000             🧑  作者: Mango

代码示例1
import torch
import torch.nn as nn

x = torch.tensor([[1.0, -1.0],
                  [0.0,  1.0],
                  [0.0,  0.0]])

in_features = x.shape[1]  # = 2
out_features = 2

m = nn.Linear(in_features, out_features)
%%%%
results would be 
>>> m.weight
tensor([[-0.4500,  0.5856],
        [-0.1807, -0.4963]])

>>> m.bias
tensor([ 0.2223, -0.6114])
%%%%