巫山巫峡

摘要： pytorch实现自定义网络层，并自设定前向传播路径-学习笔记 1. 不包含模型参数的自定义网络层 首先我们自定义一个网络层， 定义一个网络层，使其不包含模型参数，并在forward()函数中进行运算： import torch from torch import nn class non_para 阅读全文

摘要： 模型参数的初始化与访问操作 学习 如何初始化以及访问模型参数，以及如何在多层之间共享模型参数 首先定义一个含有单个隐藏层的多层感知机，使用默认方式初始化该模型的参数，并且进行一次前向计算： import torch from torch import nn from torch.nn import 阅读全文