import torch from torch import nn model = nn.Sequential(nn.Linear(2,2),nn.ReLU(),nn.Sequential(nn.Sigmoid(), nn.ReLU())) print(model)

輸出

Sequential((0): Linear(in_features=2, out_features=2, bias=True)(1): ReLU()(2): Sequential((0): Sigmoid()(1): ReLU()) )

總結

以上是生活随笔為你收集整理的输出torch构建的模型结构的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。