nn.linear的函数学习

1. nn.linear

nn.linear是线性函数，一般用于全连接层，以cnn为例，卷积层：卷积函数+激活函数relu；池化层：池化函数；全连接层：线性函数+激活函数sigmoid

 

 

in_features：输入的二维张量的大小，即输入的[ batch_size , size ]中的size；

out_features：输出的二维张量的大小，即输出的[batch_size,output_size]中的output_size；

bias：偏置参数

import torch
import torch.nn as nn

x=torch.randn(10,3,12)
print(x.size())
lin=nn.Linear(12,5)

y=lin(x)
print(y)
print(y.size())

print(lin.weight)
print(lin.weight.shape)

print(lin.bias)
print(lin.bias.shape)