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 小时前
《sklearn机器学习——数据预处理》类别特征编码
人工智能·机器学习·sklearn
FairyGirlhub2 小时前
神经网络的初始化:权重与偏置的数学策略
人工智能·深度学习·神经网络
大写-凌祁6 小时前
零基础入门深度学习:从理论到实战,GitHub+开源资源全指南(2025最新版)
人工智能·深度学习·开源·github
焦耳加热6 小时前
阿德莱德大学Nat. Commun.:盐模板策略实现废弃塑料到单原子催化剂的高值转化,推动环境与能源催化应用
人工智能·算法·机器学习·能源·材料工程
CodeCraft Studio6 小时前
PDF处理控件Aspose.PDF教程:使用 Python 将 PDF 转换为 Base64
开发语言·python·pdf·base64·aspose·aspose.pdf
深空数字孪生7 小时前
储能调峰新实践:智慧能源平台如何保障风电消纳与电网稳定?
大数据·人工智能·物联网
wan5555cn7 小时前
多张图片生成视频模型技术深度解析
人工智能·笔记·深度学习·算法·音视频
格林威8 小时前
机器视觉检测的光源基础知识及光源选型
人工智能·深度学习·数码相机·yolo·计算机视觉·视觉检测
困鲲鲲8 小时前
Python中内置装饰器
python
摩羯座-185690305948 小时前
Python数据可视化基础:使用Matplotlib绘制图表
大数据·python·信息可视化·matplotlib