pytorch_learning 4 nn.Module

torch.nn 1:简介 torch.nn是pytorch用来实现深度学习的模块,核心数据结构为nn.Module。这是一个抽象的概念,既可以表示神经网络中的一个层,也可表示包含很多层的一个网络。 2:全连接层的实现 全连接层的输出y与输入x 满足 y = Wx + b in: import torch as t from torch import nn from torch.autograd import Variable as V #继承nn.Module class Linear(nn.Module): def __init__(self,in_features,out_