Pytorch创建随机值张量的过程详解
更新时间：2023年09月14日 15:13:50 作者：知识推荐号
这篇文章主要介绍了Pytorch创建随机值张量的过程详解,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2-Pytorch创建随机值张量
1 导入必备库
import torch import numpy as np
2 使用torch.rand()创建0-1均匀分布的随机数
t = torch.rand(2,3) print(t)
输出：
tensor([[0.1452, 0.1435, 0.2124],
[0.6646, 0.5232, 0.1572]])
3 使用torch.randn()创建正态分布的随机数
t = torch.randn(2,3) print(t)
输出：
tensor([[-0.5325, -0.4938, 0.6301],
[-0.6313, -0.1412, 2.2926]])
4 使用torch.zeros()创建全0张量
t = torch.zeros(3) print(t) print(t.dtype)
输出：
tensor([0., 0., 0.])
torch.float32
5 使用torch.ones()创建全1张量
t = torch.ones(3) print(t) print(t.dtype)
输出：
tensor([1., 1., 1.])
torch.float32
6 从另一个张量创建新的张量
x = torch.zeros_like(t) print(x) x = torch.rand_like(t) print(x)
输出：
tensor([0., 0., 0.])
tensor([0.6516, 0.5740, 0.6379])
7 张量的属性：
tensor.shape返回张量的形状，与tensor.size()方法等价，tensor.dtype返回当前张量的类型
t = torch.ones(2,3,dtype= torch.float64) print(t.shape) print(t.size()) print(t.size(1)) # 返回第二维度大小 print(t.dtype) print(t.device)
输出：
torch.Size([2, 3])
torch.Size([2, 3])
3
torch.float64
cpu
8 将张量移动到显存，使用tensor.to()方法将张量移动到GPU上
# 如果GPU可用，将张量移动到显存
# 设置使用哪块芯片，多块GPU使用逗号隔开
# import os
# os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 多块GPU设置，如：'0,1,2'
if torch.cuda.is_available():
t = t.to('cuda')
print(t.device)输出：
cuda:0
9 更稳妥的移动显存方法，无论是否有GPU都能保证运行
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # 启动cuda
print("Using {} device".format(device)) # 打印当前设备
t = t.to(device)
print(t.device)输出：
Using cuda device
cuda:0
到此这篇关于Pytorch创建随机值张量的文章就介绍到这了,更多相关Pytorch随机值张量内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
无监督生成模型在机器学习中扮演着重要角色,特别是当我们在没有标签数据的情况下想要生成新的样本或理解数据的内在结构时,这篇文章主要介绍了python 无监督生成模型的示例,需要的朋友可以参考下2024-07-07
这篇文章主要介绍了tracemalloc分析内存使用情况与泄露方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-11-11
virtualenv 是一个创建隔绝的Python环境的工具。virtualenv创建一个包含所有必要的可执行文件的文件夹，用来使用Python工程所需的包。下面这篇文章就给大家介绍了python虚拟环境virualenv的安装与使用，有需要的朋友们可以参考借鉴，下面来一起看看吧。2016-12-12
这篇文章主要介绍了python 绘制国旗的示例，帮助大家利用python绘制，处理图像，感兴趣的朋友可以了解下2020-09-09
今天小编就为大家分享一篇Python计算一个点到所有点的欧式距离实现方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
这篇文章主要介绍了Python日志模块logging简介,本文讲解了Logger、Handler、Formatter、日志配置管理、通过文件配置管理日志等内容,需要的朋友可以参考下2015-04-04
本文主要介绍了Pygame如何使用精灵和碰撞检测，它们能够帮助我们跟踪屏幕上移动的大量图像。我们还会了解如何检测两个图像相互重叠或者碰撞的方法。2021-11-11
这篇文章主要介绍了Python 实现局域网远程屏幕截图案例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
这篇文章主要介绍了python实现向ppt文件里插入新幻灯片页面的方法,涉及Python操作ppt文档添加幻灯片的相关技巧,非常具有实用价值,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了Python多进程分块读取超大文件的方法,涉及Python多进程操作与文件分块读取的相关技巧,需要的朋友可以参考下2016-04-04
最新评论