PyTorch神经网络构建与直观显示

PyTorch是一个广泛使用的深度学习框架，它提供了丰富的工具和函数来构建神经网络模型。本文将介绍如何使用PyTorch构建神经网络，并展示如何直观地显示网络结构。

神经网络构建

在PyTorch中，我们可以使用torch.nn.Module类来构建神经网络。这个类提供了一些方法，例如__init__和forward，用于定义网络的结构和前向传播过程。

首先，我们需要导入必要的库：

import torch
import torch.nn as nn

接下来，我们定义一个简单的多层感知机（MLP）网络作为示例。这个网络由两个全连接层组成，每个层之间有一个ReLU激活函数。

class MLP(nn.Module