news 2026/6/10 14:30:52

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/6/10 12:44:40

LobeChat尾款催付提醒话术

LobeChat尾款催付提醒话术 在电商运营的日常中,尾款催付是一项高频但又极易出错的任务。传统方式下,客服需要手动查询订单状态、复制客户信息、套用模板生成提醒消息——流程繁琐、效率低下,还容易因语气不一致或信息遗漏引发客户不满。随着大…

作者头像 李华
网站建设 2026/6/9 21:37:54

vue基于springboot的广告服务型互联网平台

目录已开发项目效果实现截图开发技术系统开发工具:核心代码参考示例1.建立用户稀疏矩阵,用于用户相似度计算【相似度矩阵】2.计算目标用户与其他用户的相似度系统测试总结源码文档获取/同行可拿货,招校园代理 :文章底部获取博主联系方式&…

作者头像 李华
网站建设 2026/6/10 13:39:59

“AIE未来视听领袖峰会”在澳门举办,聚焦AI+AR新范式

12月4日,作为首届全球智能机械与电子产品博览会(AIE)的同期活动,“AIE未来视听领袖峰会”在澳门威尼斯人金光会展中心举行。会议以“视听全球,音画未来”为主题,汇聚300余位来自全球的行业领袖、院士专家、…

作者头像 李华
网站建设 2026/6/10 6:09:35

Linux 基础 IO 核心知识点梳理

Linux 基础 IO(输入 / 输出)是操作系统与外部设备、文件进行数据交互的核心机制,其底层围绕文件描述符展开,遵循 “一切皆文件” 的设计理念。以下是基础 IO 的核心概念与操作梳理:一、 核心概念一切皆文件Linux 中&am…

作者头像 李华
网站建设 2026/6/9 17:33:14

快速部署EmotiVoice:一键生成带情感的AI语音

快速部署EmotiVoice:一键生成带情感的AI语音 在智能语音助手越来越“懂人心”的今天,我们早已不满足于那种机械朗读式的TTS(文本转语音)。想象一下,当你的虚拟客服用带着关切语气说“您别担心”,或是游戏角…

作者头像 李华