pytorch工具——认识pytorch

目录

pytorch的基本元素操作

python 复制代码
from __future__ import print_function
import torch

创建一个没有初始化的矩阵

python 复制代码
x=torch.empty(5,3)
print(x)

创建一个有初始化的矩阵

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

创建一个全0矩阵并可指定数据元素类型为long

python 复制代码
x=torch.zeros(5,3,dtype=torch.long)
print(x)

直接通过数据创建张量

python 复制代码
x=torch.tensor([2,5,3,5])
print(x)

通过已有的一个张量创建相同尺寸的新张量

python 复制代码
x=x.new_ones(5,3,dtype=torch.double)
print(x)

利用randn_like方法得到相同尺寸张量,并且采用随机初始化的方法为其赋值

python 复制代码
y=torch.randn_like(x,dtype=torch.float)
print(y)

采用.size()方法来得到张量的形状

python 复制代码
print(x.size())

加法

第一种方法

python 复制代码
x=torch.randn(5,3)
y=torch.randn(5,3)
print(x+y)

第二种方法

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

第三种方法

python 复制代码
result=torch.empty(5,3)
torch.add(x,y,out=result)
print(result)

第四种方式:原地置换(执行的是y=y+x)

python 复制代码
y.add_(x)
print(y)


注意

切片操作

python 复制代码
x[:,1]

改变张量的形状

python 复制代码
x=torch.randn(4,4)
y=x.view(16)
z=x.view(-1,8)
x.size(),y.size(),z.size()

如果张量中只有一个元素,可以用item()将值取出,作为一个python number

python 复制代码
x=torch.randn(1)
print(x,x.item())

torch tensor和numpy array之间的互相转换

python 复制代码
a=torch.ones(5)
b=a.numpy()
a.add_(1)
print(a,b)
python 复制代码
import numpy as np
a=np.ones(5)
b=torch.from_numpy(a)
np.add(a,1,out=a)
print(a,b)


注意

关于cuda tensor:tensor可以用.to()方法将其移动到任意设备上

windows

mac

python 复制代码
if torch.backends.mps.is_available():
    device=torch.device('mps')
    #cpu上创建x,gpu上创建y
    x=torch.randn(1)
    y=torch.ones_like(x,device=device)
    x=x.to(device)
    #此时x,y都在gpu上
    z=x+y
    print(z)
    #再将z转移到cpu上
    print(z.to('cpu',torch.float32))
相关推荐
聚客AI7 分钟前
Embedding进化论:从Word2Vec到OpenAI三代模型技术跃迁
人工智能·llm·掘金·日新计划
weixin_3875456426 分钟前
深入解析 AI Gateway:新一代智能流量控制中枢
人工智能·gateway
im_AMBER35 分钟前
学习日志05 python
python·学习
大虫小呓40 分钟前
Python 处理 Excel 数据 pandas 和 openpyxl 哪家强?
python·pandas
聽雨23743 分钟前
03每日简报20250705
人工智能·社交电子·娱乐·传媒·媒体
哪 吒1 小时前
2025B卷 - 华为OD机试七日集训第5期 - 按算法分类,由易到难,循序渐进,玩转OD(Python/JS/C/C++)
python·算法·华为od·华为od机试·2025b卷
二川bro1 小时前
飞算智造JavaAI:智能编程革命——AI重构Java开发新范式
java·人工智能·重构
acstdm1 小时前
DAY 48 CBAM注意力
人工智能·深度学习·机器学习
澪-sl1 小时前
基于CNN的人脸关键点检测
人工智能·深度学习·神经网络·计算机视觉·cnn·视觉检测·卷积神经网络
羊小猪~~2 小时前
数据库学习笔记(十七)--触发器的使用
数据库·人工智能·后端·sql·深度学习·mysql·考研