news 2026/5/6 10:27:27

PyTorch 张量操作全解析:从创建到运算的完整指南

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
PyTorch 张量操作全解析:从创建到运算的完整指南

在深度学习中,张量(Tensor)是最基础也是最核心的数据结构,PyTorch 作为主流的深度学习框架,提供了丰富的张量操作 API。本文将从张量的创建、类型转换、基本运算、索引切片、形状调整到拼接等方面,通过具体示例详细讲解 PyTorch 张量的常用操作,适合初学者快速入门。

一、张量的创建

PyTorch 提供了多种创建张量的方法,可根据不同需求选择合适的方式。

1. 基于数据或形状创建

  • torch.tensor(data):根据指定数据创建张量
  • torch.Tensor(shape):根据形状创建张量(默认浮点型)
  • torch.IntTensor(shape)torch.FloatTensor(shape)等:创建指定数据类型的张量
import torch # 根据数据创建张量 print(torch.tensor([1, 2, 3])) # 输出:tensor([1, 2, 3]) # 根据形状创建张量(未初始化,值为随机) print(torch.Tensor(3, 5)) # 输出3行5列的浮点型张量 # 创建指定类型的张量(int型) print(torch.IntTensor(3, 5)) # 输出3行5列的int型张量

2. 线性与随机张量

  • torch.arange(start, end, step):创建线性序列张量
  • torch.linspace(start, end, steps):创建均分序列张量
  • torch.rand(shape):创建 [0,1) 均匀分布的随机张量
  • torch.randn(shape):创建标准正态分布的随机张量
  • torch.randint(low, high, shape):创建整数随机张量
# 设置随机种子,保证结果可复现 torch.random.manual_seed(100) # 步长为2的线性序列(0到10,不包含10) print(torch.arange(0, 10, 2)) # 输出:tensor([0, 2, 4, 6, 8]) # 0到10之间均分5个值 print(torch.linspace(0, 10, 5)) # 输出:tensor([ 0.0000, 2.5000, 5.0000, 7.5000, 10.0000]) # 3行5列的均匀分布随机张量 print(torch.rand(3, 5)) # 3行5列的标准正态分布随机张量 print(torch.randn(3, 5)) # 0-10之间的3行5列整数随机张量 print(torch.r
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/5/5 18:52:59

git 命令整理

1、git init 作用:初始化一个新的 Git 仓库,在当前目录下创建一个新的 .git 目录来管理版本历史。用法:git init 2、git clone 作用:克隆一个远程仓库到本地。用法:git clone https://github.com/username/repository.…

作者头像 李华
网站建设 2026/5/4 21:15:35

阿里健康董事长朱顺炎卸任:曾是阿里合伙人 沈涤凡接任

雷递网 乐天 12月16日阿里健康(00241.HK)日前发布公告称,公司董事长朱顺炎已辞任非执行董事、主席及提名委员会主席职务。朱顺炎已确认,彼与董事会之间并无意见分歧,亦无有关彼辞任之其他事宜须促请公司股东或联交所垂…

作者头像 李华
网站建设 2026/4/30 6:36:01

18、Mac OS X 开发中的框架与编译要点

Mac OS X 开发中的框架与编译要点 1. 框架概述 在 Mac OS X 系统中, /System/Library/Frameworks 是苹果提供的框架存放位置,这些框架的共享库可供系统上的所有应用程序使用。该目录下有三种类型的框架: - 简单公共框架 :苹果将既不是子框架也不是伞形框架的框架定义…

作者头像 李华
网站建设 2026/5/1 6:07:53

EmotiVoice在恐怖游戏音效中的惊悚表现

EmotiVoice在恐怖游戏音效中的惊悚表现 在一间昏暗的房间里,墙壁渗水,地板吱呀作响。玩家握着手电筒缓缓推进,突然,耳边传来一声低语:“你不该来的……”声音沙哑、颤抖,仿佛就在身后。你猛地回头——什么也…

作者头像 李华
网站建设 2026/5/1 6:16:26

自然语言处理在合规风险管理中的应用

自然语言处理在合规风险管理中的应用 关键词:自然语言处理、合规风险管理、文本分析、机器学习、信息提取 摘要:本文聚焦于自然语言处理(NLP)在合规风险管理领域的应用。首先介绍了研究的背景、目的、预期读者等内容,接着阐述了自然语言处理和合规风险管理的核心概念及其联…

作者头像 李华
网站建设 2026/5/3 9:37:13

148 亿赎金!近三年勒索软件狂赚,企业的钱就这么被抢了?

至少148亿元!近三年受害企业支付勒索软件赎金金额创新高 据美国财政部下属机构统计,2022-2024年期间,受害企业仅通过美国金融机构,就至少向勒索软件组织支付了超148亿元赎金,创下历史新高。 安全内参12月8日报道&…

作者头像 李华