Pytorch张量

 在conda的环境中安装Jupyter及其他软件包

Pytorch 建立在张量(tensor)之上,Pytorch张量是一个 n 维数组,类似于 NumPy 数组。专门针对GPU设计,可以运行在GPU上以加快计算效率。换句话说,Pytorch张量是可以运行在GPU上的多维数据。

 

目录

基础语法

定义张量数据类型

默认数据类型

指定数据类型

改变张量数据类型

张量与Numpy数组间的转换

将张量转换为 Numpy 数组

将 Numppy 数组转换为张量

在设备间移动张量

默认在CPU上定义张量

定义CUDA设备

将张量移动到CUDA设备

将张量移动到CPU

在任意设备上定义张量


基础语法

Pytorch张量

Numpy数组

torch.ones(.)numpy.ones(.) 创建一个一数组
torch.zeros(.) numpy.zeros(.)创建一个零数组
 torch.rand(.)numpy.random.rand(.)创建随机数组
torch.tensor(.)numpy.array(.) 从给定值创建数组
x.shape or x.size()x.shape获取数组形状

定义张量数据类型

默认数据类型

默认张量数据类型为 torch.float32

import torch
x = torch.ones(2, 2)
print(x)
print(x.dtype)

 

指定数据类型

x = torch.ones(2, 2, dtype=torch.int8)
print(x)
print(x.dtype)

改变张量数据类型

先定义一个张量,打印其数据类型

x=torch.ones(1)
print(x.dtype)

 通过".type()"命令改变数据类型

x=x.type(torch.torch.uint8)
print(x.dtype)

 可以看到数据类型由torch.float32改变为了torch.uint8

张量与Numpy数组间的转换

将张量转换为 Numpy 数组

 定义一个随机张量

x=torch.rand(2,2)
print(x)
print(x.dtype)

 

通过".numpy()"命令转换为numpy数组

y=x.numpy()
print(y)
print(y.dtype)


将 Numppy 数组转换为张量

定义一个numpy数组

import numpy as np
x=np.zeros((2,2),dtype=np.float32)
print(x)
print(x.dtype)

用".from_numpy()"命令转换为张量

y=torch.from_numpy(x)
print(y)
print(y.dtype)

在设备间移动张量

一般情况下,Pytorch 张量存储在 CPU 上,张量运行在 GPU上 以加快计算速率,这是张量的主要优势。为了有效利用这个优势,需要将张量移动到 CUDA 设备,可以使用 .to 方法将张量移动到相应设备上。 

默认在CPU上定义张量

使用".tensor()"命令,张量一般默认定义在CPU上

x=torch.tensor([1, 1.5])
print(x)
print(x.device)

 

 

定义CUDA设备

if torch.cuda.is_available():device=torch.device("cuda:0")
torch.cuda.current_device()

 

 

将张量移动到CUDA设备

x = x.to(device)
print(x)
print(x.device)

 

 

将张量移动到CPU

device = torch.device("cpu")
x = x.to(device)
print(x)
print(x.device)

 

在任意设备上定义张量

device = torch.device("cuda:0")
x = torch.ones(2,2, device=device)
print(x)

 

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/bicheng/46252.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

wget介绍

wget介绍 wget是一个广泛使用的命令行工具,主要用于从网络上自动下载文件。它支持HTTP、HTTPS和FTP等多种协议,具有递归下载、断点续传、限速下载等功能,使其成为一个强大而灵活的下载工具。以下是对wget的详细介绍和使用方法的概述&#xf…

解决QT creator中文乱码问题

1.首先设置文本编辑器为UTF-8 先在工具-选项-文本编辑器-behavior部分选择文件编码为UTF-8,紧接着是选择“如果编码是UTF-8则添加”,如下图 2.设置ext code for tools 为system 具体解决办法是 工具-选项-环境-interfaces这一栏有一个“Text code for to…

最新JJWT 0.12.6学习

本文参考的是jjwt官方github,链接在此,本文会持续跟进jjwt的最新版本 GitHub - jwtk/jjwt: Java JWT: JSON Web Token for Java and Android 简介 JJWT(Java JWT)是Java平台上相当流行的用于生成Json Web Token的库&#xff0c…

DP讨论——桥接模式

学而时习之,温故而知新。 敌人出招(使用场景) 客户不断变化的需求是敌人,加个需求,再加个需求,然后内部实现不断新增类。 你出招 采用桥接模式。 一个是桥梁对象,另一个是业务实现类。客户通…

云计算安全需求分析与安全保护工程

云计算基本概念 云计算(Cloud Computing)是一种通过互联网提供计算资源和服务的技术。它允许用户按需访问和使用计算资源,如服务器、存储、数据库、网络、安全、分析和软件应用等,而无需管理底层基础设施。以下是云计算的基本概念…

快捷:通过胶水语言实现工作中测试流程并行、加速

通过胶水语言实现工作中测试流程并行、加速 通过胶水语言实现工作中测试流程并行、加速工作场景(背景)问题抽象(挑战)如何做(行动)获得了什么(结果)后记相关资源 通过胶水语言实现工…

代码随想录(day6)哈希表-求两个数组的交集

什么是红黑树,参考链接【数据结构】史上最好理解的红黑树讲解,让你彻底搞懂红黑树-CSDN博客 题目: 注意:字典用{}符合、元组用()符号、列表用[]符号 table.get(num,0)1的意思 class Solution(object):def intersection(self, n…

qt 用代码添加 QPushButton

在 Qt 中使用代码添加 QPushButton 控件是一个基本的操作。以下是一个简单的例子&#xff0c;展示了如何在 Qt 的一个继承自 QWidget 或 QMainWindow 的类中添加一个 QPushButton&#xff1a; 首先&#xff0c;你需要包含必要的头文件&#xff1a; #include <QWidget> /…

【C++】——入门基础

文章目录 命名空间输入与输出缺省参数函数重载引用内敛函数指针控制nullptr 命名空间 当C语言遇到命名冲突时&#xff0c;很难解决&#xff0c;所以C关键字namespace就是针对这种问题的。 如何使用命名空间内的成员呢&#xff1f; 加命名空间名称及限定符&#xff08; &#…

JS【详解】数据类型转换

数值 → 其他数据类型 let num 123;数值 → 字符串 结果为由该数值组成的字符串 String(num) // 结果为 123 String(NaN) // 结果为 NaNnum num.toString(); 数值 → 布尔值 0 , -0 , NAN 转为布尔值后为 false其他数值转为布尔值后为 true Boolean(num )// 取反再取反 …

husky 和 lint-staged 构建代码项目规范

目录 前言 最简单的方法 过 scripts 来解决如果检测工具多&#xff0c;需要多次处理 通过 husky(哈士奇)来解决容易遗忘的问题 1. 安装 2. husky init 3. 试一试​ lint-stadge 只 lint 改动的 1. 安装 2. 修改 package.json 配置 3. 添加 npm 脚本: 4.使用 Husky…

MAVSDK动态库与静态库及mavsdk_server程序macOS平台编译与安装

1.克隆mavsdk: git clone https://github.com/mavlink/MAVSDK.git --recursive 2.编译静态库 cmake -Bbuild/default -H. -DCMAKE_BUILD_TYPE=Release -DBUILD_SHARED_LIBS=OFF 生成makefile 生成成功,开始编译 cmake --build build/default -j8 成功生成libmavsdk.a 开…

Java之Stream流的笔记--手写版

Stream流通过讲集合或数组转换成链状流式的结构&#xff0c;简化了集合和数组进行排序、筛选、遍历、去重、统计等操作。主要包括创建流、中间操作、终结操作。若流中无终结操作&#xff0c;则中间操作不会执行&#xff1b;流是一次性的&#xff0c;使用完就会失效&#xff0c;…

vmware workstation 虚拟机安装

vmware workstation 虚拟机安装 VMware Workstation Pro是VMware&#xff08;威睿公司&#xff09;发布的一代虚拟机软件&#xff0c;中文名称一般称 为"VMware 工作站".它的主要功能是可以给用户在单一的桌面上同时运行不同的操作系统&#xff0c;它也是可进 行开发…

Jump Point Search (JPS) 路径规划算法

Jump Point Search (JPS) 是一种优化的路径规划算法,主要用于加速A*搜索算法在网格地图上的性能。它通过减少要评估的节点数量来实现这一目标。以下是对JPS算法的详细解释: 基本思想 JPS利用了路径规划中的对称性,跳过不必要的节点,直接跳到关键节点(即“跳点”)。这些…

HDFS体系架构文件写入/下载流程

HDFS体系架构 HDFS&#xff08;Hadoop Distributed File System&#xff0c;Hadoop分布式文件系统&#xff09;是Hadoop项目中的一个核心组件&#xff0c;旨在以高容错、高吞吐量来处理大规模数据集。它的体系架构由以下几个主要部分组成&#xff1a;Client&#xff0c;NameNo…

去中心化预言机是什么

去中心化预言机&#xff08;Decentralized Oracle&#xff09;是区块链技术中的一个组件&#xff0c;它允许智能合约访问链外&#xff08;off-chain&#xff09;的数据源。智能合约是自动执行的合约条款&#xff0c;它们在区块链上运行&#xff0c;但它们本身无法直接访问链外的…

PDF公式转Latex

文章目录 摘要数据集 UniMER介绍下载链接 LaTeX-OCRUniMERNet安装UniMER 用的数据集介绍下载链接 PDF-Extract-Kit整体介绍效果展示评测指标布局检测公式检测公式识别 使用教程环境安装参考[模型下载](models/README.md)下载所需模型权重 在Windows上运行在macOS上运行运行提取…

易备数据备份软件,全方位的防勒索数据备份方案

企业 IT 基础设施的潜在物理故障、系统故障、未及时修补的系统及网络安全漏洞、人为误操作&#xff08;误删除及误点击钓鱼邮件等&#xff09;、勒索软件的无孔不入、自然灾害的威胁等&#xff0c;就像若干不定时炸弹&#xff0c;为企业数据资产的安全埋下了巨大的引爆风险。 …

MySQL:库操作

1. 创建数据库 create database [if not exists] name [create_specification], [create_specification]... []内为可选的选项 create_specification: character set charset_name -- 指定数据库采用的字符集 -- 数据库未来存储数据 collate collation_name -- 指定数据库字符…