pytorch 使用tensorboardX可视化网络结构图 前提条件 使用方法 代码 使用 结果展示: 前提条件 安装tensorflow,tensorboardX,pytorch 使用方法 代码 from tensorboardX import SummaryWriter mynetwork = MyNetWork() # 定义的网络类名 input_ = torch.randn((1, 3, 224, 224)