从PyTorch官方的一篇教程说开去(6.1 - 张量 tensor 基本操作)

您的进步和反馈是我最大的动力,小伙伴来个三连呗!共勉。

其实深度学习的框架,出品最早的是谷歌的TensorFlow,我也有在前面的文章中有提到过G家的开创性,感兴趣的可以移步去看下 - 从PyTorch官方的一篇教程说开去(1 - 初心)-CSDN博客

但是事易时移,脸书家的PyTorch靠着python社区,在开发群体中变得更加流行。当然了,马斯克也献上了助攻。我们本篇只讨论应用,所以按下这个有点八卦的话头不提。

吐槽一下,实际上两个框架的底层都是C++,而不是python,从后面的输出风格就看的出来。而在了解和迎合学术界和社区方面,脸书是职业本能,确实硬压了G家的直男们一头。

人工智能的基础是矩阵,是向量(也就是本文主题的Tensor),我们在前篇中有展示过AI的大脑的样子。有兴趣的请看前面的文章 - 从PyTorch官方的一篇教程说开去(4 - Q-table来源及解决问题实例)-CSDN博客

为了走的更远,大家一起给"张量"这个词补补课哈。

方便展示,请在Jupiter中运行本篇中的代码。

1 - 创建原始数据:

python 复制代码
import torch
import numpy as np

data = [[1,2],[3,4]]
print(data)
  • 这部分创建了一个 Python 列表,其中包含两个列表,用于模拟一个 2x2 的矩阵。

2 - 转换为 PyTorch 张量:

python 复制代码
x_data = torch.tensor(data)
print(x_data)
  • 列表转换为一个张量

3 - 创建相同形状的全1张量:

python 复制代码
x_ones = torch.ones_like(x_data)
print(f"Ones Tensor: \n {x_ones} \n")

4 - 创建具有相同形状的随机张量:

python 复制代码
x_rand = torch.rand_like(x_ones, dtype=torch.float)
print(f"Random Tensor: \n {x_rand} \n")

5 - 创建不同形状的全0、全1和随机张量:

python 复制代码
shape = (4,4)
rand_tensor = torch.rand(shape)
ones_tensor = torch.ones(shape)
zeros_tensor = torch.zeros(shape)

print(f"Random Tensor: \n {rand_tensor} \n")
print(f"Ones Tensor: \n {ones_tensor} \n")
print(f"Zeros Tensor: \n {zeros_tensor} \n")

print (f"{rand_tensor.shape},\n{rand_tensor.dtype},\n{rand_tensor.device}")

6 - 索引,切片和赋值操作:

python 复制代码
print(f"{ones_tensor[0]}\n")
print(f"{ones_tensor[:,0]}\n")
print(f"{ones_tensor[...,-1]}\n")
ones_tensor[:,1] = 0
print(f"{ones_tensor}\n")

7 - 张量的拼接操作:

python 复制代码
t1 = torch.cat([ones_tensor,ones_tensor,ones_tensor],dim=1)
print(t1)

该吃饭去了,未完,下一节继续哈~

相关推荐
kisshuan12396几秒前
电信杆塔类型识别与分类_fovea_r101_fpn_4xb4-2x_coco模型详解_模型训练与验证_通俗易懂!入门必看系列!
人工智能·目标跟踪·分类
子午1 分钟前
【2026原创】动物识别系统~Python+深度学习+人工智能+模型训练+图像识别
人工智能·python·深度学习
o_insist7 分钟前
LangChain1.0 实现 PDF 文档向量检索全流程
人工智能·python·langchain
OpenMiniServer9 分钟前
AI + GitLab + VSCode:下一代开发工作流的革命性集成
人工智能·vscode·gitlab
脑洞AI食验员12 分钟前
智能体来了:用异常与文件处理守住代码底线
人工智能·python
摘星观月15 分钟前
【三维重建2】TCPFormer以及NeRF相关SOTA方法
人工智能·深度学习
shangjian00716 分钟前
AI大模型-机器学习-分类
人工智能·机器学习·分类
Tiny_React18 分钟前
使用 Claude Code Skills 模拟的视频生成流程
人工智能·音视频开发·vibecoding
人工小情绪20 分钟前
深度学习模型部署
人工智能·深度学习
Codelinghu22 分钟前
「 LLM实战 - 企业 」构建企业级RAG系统:基于Milvus向量数据库的高效检索实践
人工智能·后端·llm