当前位置:网站首页>Pytorch使用和技巧
Pytorch使用和技巧
2022-08-05 00:27:00 【流黄蛋】
#先导入包
import torch
import numpy as np
1. tensor的创建
1.1 初始化tensor的方法
data = [[1, 2][3, 4]] # 使用列表初始化tensor
tensor_data = torch.tensor(data)
# 通过 numpy数组创建tensor
data = np.random.normal((2, 3))
tensor_data = torch.tensor(data)
# 创建一个和另一个张量大小相同的全一tensor
data_ones = torch.ones_like(data)
# 创建一个和另一个张量大小相同的随机张量
data_rand = torch.rand_like(data)
# 创建随机张量
rand_data = torch.rand([2, 3]) # 其中的size可以是数组,tuple,不能为字典
1.2 查看tensor属性
data = torch.randn((2, 3))
# 判断是否是张量
data.is_tensor() # return True or False
data.is_complex(). # 数据是复数返回True, 否则返回False
data.is_floating_point() # 数据为浮点类型返回True, 否则为False
torch.is_nonzero(data) # 是不是包含单一元素tensor,并且如果是0就返回False, 否则就是True
torch.numel(data) # 判断张量中所有元素的数目
1.3 张量操作
torch.zeros([H, W]) # 返回全零张量
torch.range(start=0, end, step=1) # 生成连续的张量, 传入一个参数默认是end
torch.arange(start, end, step=1) # 也是生成连续的张量
# 注意torch.range生成的连续数组比torch.arange要多一个元素
torch.eye(n, m=None) # 创建一个 n * m 的单位矩阵,没有m就会创建 n * n的方阵
torch.full([H, W], m) , 创建一个H * W,元素全部为m的矩阵
torch.cat()# 对张量进行连接
# 例如
使用tensorflow时候没有考虑过这个事情,今天读代码发现torch还有这样一个重要的语句,记录学习一下:
1. model.train()
启用 BatchNormalization 和 Dropout
2. model.eval()
不启用 BatchNormalization 和 Dropout
训练完 train 样本后,生成的模型 model 要用来测试样本。在 model(test) 之前,需要加上model.eval(),否则只要有输入数据,即使不训练,model 也会改变权值。这是model中含有的 batch normalization 层所带来的的性质。
转载自:https://zhuanlan.zhihu.com/p/208233193
边栏推荐
猜你喜欢
[Cloud Native--Kubernetes] Pod Controller
【LeetCode】Summary of Two Pointer Problems
redis可视化管理软件Redis Desktop Manager2022
【Valentine's Day special effects】--Canvas realizes full screen love
软件开发工具的技术要素
matlab中rcosdesign函数升余弦滚降成型滤波器
[230]连接Redis后执行命令错误 MISCONF Redis is configured to save RDB snapshots
MAUI Blazor 权限经验分享 (定位,使用相机)
Mysql_14 存储引擎
could not build server_names_hash, you should increase server_names_hash_bucket_size: 32
随机推荐
ansible学习笔记分享-含剧本示例
oracle创建用户
2022牛客多校训练第二场 J题 Link with Arithmetic Progression
[Cloud Native--Kubernetes] Pod Controller
软件测试面试题:请你分别画出 OSI 的七层网络结构图和 TCP/IP 的四层结构图?
Software Testing Interview Questions: What's the Key to a Good Test Plan?
软件测试面试题:您以往所从事的软件测试工作中,是否使用了一些工具来进行软件缺陷(Bug)的管理?如果有,请结合该工具描述软件缺陷(Bug)跟踪管理的流程?
2022杭电多校训练第三场 1009 Package Delivery
找不到DiscoveryClient类型的Bean
oracle创建表空间
could not build server_names_hash, you should increase server_names_hash_bucket_size: 32
Software testing interview questions: What are the seven-layer network protocols?
电子行业MES管理系统的主要功能与用途
leetcode:269. 火星词典
More than 2022 cattle school training topic Link with the second L Level Editor I
怎样进行在不改变主线程执行的时候,进行日志的记录
在线中文姓名生成工具推荐
2022牛客多校第三场 A Ancestor
matlab中rcosdesign函数升余弦滚降成型滤波器
MAUI Blazor 权限经验分享 (定位,使用相机)