上面的代码定义了一个类Net,继承了torch.nn.Module。在Net的__init__()函数中定义了卷积层conv1和conv2,池化层pool,全连接层fc1、fc2、fc3。并在forward函数中定义了一个前向传播过程。在代码最后,使用了torch.randn来生成一个随机张量,并使用自定义的网络进行一次前向传播,最后使用torchviz中的make_dot函数可视化网络结构,并保存为.png文件。
import torch import torch.nn as nn from torchviz import make_dot import torch.nn.functional as F cl...
最新推荐文章于 2024-04-28 13:47:55 发布