pytorch学习笔记十一：查看模型参数信息的几种方式

首先，我们先创建一个只有两层的简单的模型，第一层是没有参数的FlattenLayer,第二层是有参数的Linear.

import torch
from torch import nn

num_inputs = 5
num_hidden =  2
num_outputs = 1

class FlattenLayer(nn.Module):
    def __init__(self)