Breaking News: Grepper is joining You.com. Read the official announcement!

torch.nn.Linear(in_features, out_features, bias=True) discription

Terrible Trout answered on July 17, 2020 Popularity 7/10 Helpfulness 1/10

answer torch.nn.Linear(in_features, out_features, bias=True) discription

torch.nn.Linear(in_features, out_features, bias=True) discription

Comment

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

Popularity 7/10 Helpfulness 1/10 Language python

Source: stackoverflow.com

Tags: python

Link to this answer
Share Copy Link