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))
相关推荐
张较瘦_1 小时前
[论文阅读] 人工智能 + 软件工程 | 需求获取访谈中LLM生成跟进问题研究:来龙去脉与创新突破
论文阅读·人工智能
一 铭2 小时前
AI领域新趋势:从提示(Prompt)工程到上下文(Context)工程
人工智能·语言模型·大模型·llm·prompt
云泽野5 小时前
【Java|集合类】list遍历的6种方式
java·python·list
麻雀无能为力5 小时前
CAU数据挖掘实验 表分析数据插件
人工智能·数据挖掘·中国农业大学
时序之心6 小时前
时空数据挖掘五大革新方向详解篇!
人工智能·数据挖掘·论文·时间序列
IMPYLH6 小时前
Python 的内置函数 reversed
笔记·python
.30-06Springfield6 小时前
人工智能概念之七:集成学习思想(Bagging、Boosting、Stacking)
人工智能·算法·机器学习·集成学习
说私域7 小时前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的超级文化符号构建路径研究
人工智能·小程序·开源
永洪科技7 小时前
永洪科技荣获商业智能品牌影响力奖,全力打造”AI+决策”引擎
大数据·人工智能·科技·数据分析·数据可视化·bi
shangyingying_17 小时前
关于小波降噪、小波增强、小波去雾的原理区分
人工智能·深度学习·计算机视觉