- 博客(2)
- 收藏
- 关注
原创 Pytorch学习笔记2 网络的定义与优化
网络的定义与优化 一 PyTorch中定义网络的方式 PyTorch中提供了多种搭建网络的方法,下面以一个简单的全连接神经网络为例,介绍定义网络的过程。我们将使用Module和Sequential两种不同的方式定义网络 首先导入需要的相应模块 import torch import torch.nn as nn import torch.optim as optim import torch.utils.data as Data from sklearn.datasets import load_b
2021-07-22 17:11:38 513
原创 Pytorch学习笔记1
Pytorch学习笔记1 导入pytorch模块,若没有安装pytorch,可到 pytorch安装选择相应系统版本进行安装 import torch import numpy as np tensor的基本属性与生成 #生成2*3的随机张量 a = torch.FloatTensor(2,3) #生成默认类型的2*3的张量通过dtype查看类型 b = torch.Tensor(2,3) print(a) print(b.dtype) a = torch.tensor([4]) print(a) #
2021-07-22 11:05:42 117
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人