Pytorch——nn.Flatten()函数

William.csj

已于 2023-12-22 10:51:57 修改

阅读量681

点赞数 1

分类专栏： PyTorch 文章标签： pytorch 人工智能 python

于 2023-12-22 10:50:48 首次发布

原文链接：https://blog.csdn.net/kuan__/article/details/116987162

版权

PyTorch 专栏收录该内容

39 篇文章 1 订阅

订阅专栏

nn.Flatten()：对图像数组进行展平操作

例子

# nn.Flatten()：对图像数组进行展平操作
net = nn.Sequential(nn.Flatten(),
                    nn.Linear(784, 256),
                    nn.ReLU(),
                    nn.Linear(256, 10))

def init_weights(m):
    if type(m) == nn.Linear:
        nn.init.normal_(m.weight, std=0.01) # 给线性层的权重初始化，符合正态分布，标准差为0.01

net.apply(init_weights);