51工具盒子

依楼听风雨
笑看云卷云舒,淡观潮起潮落

linear

PyTorch 中 Linear 和 Embedding 网络层

PyTorch 中 Linear 和 Embedding 网络层

厉飞雨 阅读(7) 评论(0) 赞(1)

当我们使用 PyTorch 构建神经网络时,经常使用到一些内置的网络层。本篇文章主要介绍下列层的使用: 1. 线性层(Linear) 2. 词嵌入层(Embedding) 3. 循环网络层(RNN、GRU、LSTM) 1. 线性层 {#title-0} ================= ``` torch.nn.Linear(in_features, out_featu...