pytorch代码复现1(基础知识)

创建矩阵

全零矩阵

复制代码
In [4]:
import torch
torch.__version__
x=torch.empty(5,3)
x

Out[4]:

复制代码
tensor([[0.0000e+00, 0.0000e+00, 4.6430e-23],
        [1.4013e-45, 1.2612e-44, 0.0000e+00],
        [3.5733e-43, 0.0000e+00, 0.0000e+00],
        [0.0000e+00, 0.0000e+00, 0.0000e+00],
        [0.0000e+00, 0.0000e+00, 0.0000e+00]])

随机矩阵

In [5]:

复制代码
x=torch.rand(5,3)
x

Out[5]:

复制代码
tensor([[0.8045, 0.6600, 0.5920],
        [0.9726, 0.2459, 0.5417],
        [0.5958, 0.6286, 0.5736],
        [0.5969, 0.0276, 0.8971],
        [0.9583, 0.4394, 0.5928]])

#tensor(张量)几维矩阵都行

初始化一个全零矩阵

In [20]:

复制代码
x=torch.zeros(5,3)#x=torch.zeros(5,4,dtype=torch.int)
x

Out[20]:

复制代码
tensor([[0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.]])

直接传入数据

In [18]:

复制代码
y=torch.tensor([5,4])
y

Out[18]:

复制代码
tensor([5, 4])

显示矩阵大小

In [21]:

复制代码
x.size()#当前的维度是几行几列的

Out[21]:

复制代码
torch.Size([5, 3])

矩阵相加

法1:

In [25]:

复制代码
y=torch.rand(5,3)
x=torch.rand(5,3)
x+y

Out[25]:

复制代码
tensor([[0.8520, 0.6184, 1.2141],
        [1.8745, 1.0329, 1.1968],
        [0.9743, 0.5262, 1.4275],
        [0.5415, 1.0113, 1.2635],
        [0.9762, 0.7496, 1.4369]])

法2:

In [26]:

复制代码
torch.add(x,y)

Out[26]:

复制代码
tensor([[0.8520, 0.6184, 1.2141],
        [1.8745, 1.0329, 1.1968],
        [0.9743, 0.5262, 1.4275],
        [0.5415, 1.0113, 1.2635],
        [0.9762, 0.7496, 1.4369]])

索引

In [27]:

复制代码
x[:1]

Out[27]:

复制代码
tensor([[0.0229, 0.1664, 0.5243]])

改变矩阵维度

In [28]:

复制代码
x=torch.rand(4,4)
y=x.view(16)
z=x.view(-1,8)#-1代表自动填充
print(x.size(),y.size(),z.size())
复制代码
torch.Size([4, 4]) torch.Size([16]) torch.Size([2, 8])

tensor转成numpy的格式

In [30]:

复制代码
a=torch.ones(5)
b=a.numpy()
b

Out[30]:

复制代码
array([1., 1., 1., 1., 1.], dtype=float32)

numpy转tensor的格式

In [29]:

复制代码
import numpy as np
a=np.ones(5)
b=torch.from_numpy(a)
b

Out[29]:

复制代码
tensor([1., 1., 1., 1., 1.], dtype=torch.float64)
相关推荐
mortimer2 小时前
Python 文件上传:一个简单却易犯的错误及解决方案
人工智能·python
IT_陈寒2 小时前
Vue3性能优化实战:这5个技巧让我的应用加载速度提升了70%
前端·人工智能·后端
机器之心2 小时前
英伟达50亿美元入股英特尔,将发布CPU+GPU合体芯片,大结局来了?
人工智能·openai
新智元2 小时前
芯片大地震,黄仁勋355亿入股!英特尔要为老黄造CPU,股价狂飙30%
人工智能·openai
阿然1653 小时前
首次尝试,95% 的代码都是垃圾:一位工程师使用 Claude Code 六周的心得
人工智能·agent·ai编程
martinzh3 小时前
RAG系统优化大揭秘:让你的AI从学渣变学霸的进化之路
人工智能
汀丶人工智能3 小时前
想成为AI绘画高手?打造独一无二的视觉IP!Seedream 4.0 使用指南详解,创意无界,效率翻倍!
人工智能
蚝油菜花3 小时前
万字深度解析Claude Code的Hook系统:让AI编程更智能、更可控|下篇—实战篇
人工智能·ai编程·claude
中杯可乐多加冰4 小时前
从创意到应用:秒哒黑客松大赛 用零代码点燃你的创新火花
人工智能
百度Geek说4 小时前
一文解码百度地图AI导航“小度想想”
人工智能