百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术分类 > 正文

pytorch的基础使用(一)

ztj100 2025-02-11 14:27 38 浏览 0 评论

本文将从最基础的 pytorch 操作一步步学习

1、通过python列表创建 torch数据

# 导入pytorch的库在这里也就是torch
import torch
# 通过一个列表来创建 torch
data1 = torch.tensor([[1.1, -1.1], [1. , -1.1]])

这里备注下,其实在 torch 中 tensor 很多操作都很类似 numpy 这个库

2、通过 numpy 的数组来创建 tensor

data2 = torch.tensor(np.array([[1,2,3],[4, 5, 6]]))

输出:
tensor([[1, 2, 3],
        [4, 5, 6]], dtype=torch.int32)

3、通过 torch自带的 api 来创建 tensor

torch.empty(3, 4 )  创建3行4列的空tensor,说是创建空的,其实就是0
torch.ones( [3,4] ) 创建3行4列的全为1的 tensor
torch.zeros([3,4]) 创建3行4列全为0的tensor
torch.rand([3,4]) 创建一个3行4列的tensor,其中的元素为0~1 之间的数据
torch.randn([3,4]) 创建一个3行4列的tensor,其实里面的元素方差为1,均值为0

4、获取 torch 中的数据

4.1 当只有一个元素的时候

  	# 当 torch 中只有一个元素的时候,可以用 item() 来获取这个元素
	  t1 = a = torch.tensor(np.arange(1))
    print(t1)
    # out:tensor([0], dtype=torch.int32)
    print(t1.item())
    # out:0

4.2 转化成 numpy 数组来实现读取。

    t1 = a = torch.randn([3, 4])
    print(type(t1))
    print(t1)
    print(type(t1.numpy()))
    print(t1.numpy())
out:

tensor([[-2.1239, -0.0909, -1.5348, -1.2876],
        [-0.9081,  0.3360,  1.6969,  0.3123],
        [ 0.3102,  0.8689, -0.3897,  0.2151]])

[[-2.123888   -0.09085716 -1.534757   -1.2876265 ]
 [-0.90809774  0.3359716   1.696873    0.3122723 ]
 [ 0.31016025  0.86885756 -0.38969612  0.21510045]]

从上面输出的结果来看,torch 经过转换后就变成了 numpy 这个数组

5、torch 其他常见的属性操作

5.1、torch 获取其形状

data2 = torch.empty(3, 4)
data2.size()

# torch.Size([3, 4])

5.2、torch 获取其形状

data2 = torch.empty(3, 4)
data2 = data2.view(2,6)

# tensor([[0., 0., 0., 0., 0., 0.],
        [0., 0., 0., 0., 0., 0.]])

在 numpy 中改变形状是使用 shape,来改变,而在 torch 中是通过 view 来改变

5.3、其他操作

获取最大值:`tensor.max()`
转置:`tensor.t()`
获取某行某列的值:`tensor[1,3]`  这里就是获取tensor中第一行第三列的值
tensor[1,3]=100` 对tensor中第一行第三列的位置进行赋值100`
 torch切片: x[:,1]

6、torch 常见的数据类型

数据类型的获取   x.dtype  # torch.int32
数据类型的设置   torch.ones([2,3],dtype=torch.float32)
数据类型的修改  a.type(torch.float)  # 比如a 在之前已经设置过了,如果这样再设置一遍就是修改了

7、torch 的基本运算操作

x = x.new_ones(5, 3, dtype=torch.float)
 y = torch.rand(5, 3)
# 相加操作
 x+y
 torch.add(x,y)
x.add(y)
x.add_(y)
注意的是最后这一种操作会改变X的值 
x +10  # 也就是对每个元素进行+10

相关推荐

Docker安全开放远程访问连接权限(docker 远程授权访问)

1、Docker完全开放远程访问Docker服务完全开放对外访问权限操作如下:#开启端口命令(--permanent永久生效,没有此参数重启后失效)firewall-cmd--zone=pu...

SpringCloud系列——4OpenFeign简介及应用

学习目标什么是OpenFeign以及它的作用RPC到底怎么理解OpenFeign的应用第1章OpenFeign简介在前面的内容中,我们分析了基于RestTemplate实现http远程通信的方法。并...

Spring Boot集成qwen:0.5b实现对话功能

1.什么是qwen:0.5b?模型介绍:Qwen1.5是阿里云推出的一系列大型语言模型。Qwen是阿里云推出的一系列基于Transformer的大型语言模型,在大量数据(包括网页文本、书籍、代码等)...

JDK从8升级到21的问题集(jdk8升级到11)

一、背景与挑战1.升级动因oOracle长期支持策略o现代特性需求:协程、模式匹配、ZGC等o安全性与性能的需求oAI新技术引入的版本要求...

大白话详解Spring Cloud服务降级与熔断

1.Hystrix断路器概述1.1分布式系统面临的问题复杂分布式体系结构中的应用程序有数十个依赖关系,每个依赖关系在某些时候将不可避免地失败。这就造成有可能会发生...

面试突击43:lock、tryLock、lockInterruptibly有什么区别?

在Lock接口中,获取锁的方法有4个:lock()、tryLock()、tryLock(long,TimeUnit)、lockInterruptibly(),为什么需要这么多方法?这些方法都有...

了解网络编程 TCP/IP 协议与UDP 协议

因为iP地址比较难记忆,很多情况下可以使用域名代替iP地址。1.TCP/IP协议与UDP协议通过IP地址与端口号确定计算机在网络中的位置后,接下来考虑通讯的问题:因为不同计算机的软硬件平台...

Semaphore与Exchanger的区别(semaphore和signal)

Semaphore和Exchanger是Java并发编程中两个常用的同步工具类,它们都可以用于协调多个线程之间的执行顺序和状态,但它们的作用和使用方式有所不同:Semaphore类表示一个...

Java教程:什么是分布式任务调度?怎样实现任务调度?

通常任务调度的程序是集成在应用中的,比如:优惠卷服务中包括了定时发放优惠卷的的调度程序,结算服务中包括了定期生成报表的任务调度程序...

java多线程—Runnable、Thread、Callable区别

多线程编程优点:进程之间不能共享内存,但线程之间共享内存非常容易。系统创建线程所分配的资源相对创建进程而言,代价非常小。Java中实现多线程有3种方法:继承Thread类实现Runnable...

工厂模式详解(工厂模式是啥意思)

工厂模式详解简单工厂简单工厂模式(SimpleFactoryPattern)是指由一个工厂对象决定创建出哪一种产品类的实例。简单工厂适用于工厂类负责创建的对象较少的场景,且客户端只需要传入工厂类的...

我们程序员眼中的母亲节(你眼中的程序员是什么样子的?程序员的薪酬如何?)

导语:对于我们成人来说,尤其是漂泊在外的程序员,陪伴父母的时间太少了。每逢佳节倍思亲,我们流浪外在的游子应该深有感触。母亲,是世界上最伟大的人,她承载着对我们的爱,更是负担和压力。我们作为子女,只会嫌...

死锁的 4 种排查工具(死锁检测方法要解决两个问题)

死锁(DeadLock)指的是两个或两个以上的运算单元(进程、线程或协程),都在等待对方停止执行,以取得系统资源,但是没有一方提前退出,就称为死锁。死锁示例接下来,我们先来演示一下Java中最简...

1. 工厂模式详解(工厂模式示例)

我们的项目代码也是由简而繁一步一步迭代而来的,但对于调用者来说却是越来越简单化。简单工厂模式简单工厂模式(SimpleFactoryPattern)是指由一个工厂对象决定创建出哪一种产品类的实例。...

Jmeter(二十):jmeter对图片验证码的处理

jmeter对图片验证码的处理在web端的登录接口经常会有图片验证码的输入,而且每次登录时图片验证码都是随机的;当通过jmeter做接口登录的时候要对图片验证码进行识别出图片中的字段,然后再登录接口中...

取消回复欢迎 发表评论: