nn.Linear函数形参有哪些
时间: 2023-10-26 14:37:57 浏览: 79
函数指针、函数名、二维数组指针做形参
nn.Linear函数的形参如下:
- in_features: 输入的特征数量
- out_features: 输出的特征数量
- bias: 是否在计算中使用偏差,默认为True,表示使用偏差
- device: 用于指定在哪个设备上创建模型参数,默认为None,表示使用当前设备
- dtype: 用于指定模型参数的数据类型,默认为None,表示使用默认数据类型
阅读全文