【PyTorch学习笔记】一、两个全连接层的感知机实现(损失函数及优化器的使用)
import torch
from torch import nn
# 全连接类
class Linear(nn.Module):
def __init__(self, in_dim, out_dim):
super(Linear, self).__init__()
# Parameter 为Tensor的构造方法,默认求导
# w b 为网络需要学习的参数
self.w = nn.Parameter(torch.randn(in_d.
原创
2020-05-31 21:49:44 ·
1744 阅读 ·
5 评论