第一章 pytorch(入门-张量)

本文作为PyTorch入门介绍,详细阐述PyTorch与numpy的对比及转换,并重点讲解了PyTorch中张量的学习,包括常用函数如tensor.mean()和torch.argmax()的用法,以及如何创建不同类型的张量。
摘要由CSDN通过智能技术生成

第一章 pytorch(入门)

PyTorch是一个基于 Python 的科学计算包,它主要有两个用途:
1.类似Numpy但是能利用GPU加速
2.一个非常灵活和快速的用于深度学习的研究平台

numpy与pytorch对比与转换

pytorch 中常用的函数

1.tensor.mean()
tensor.mean()
dim=0表示计算第一维度,等于1等于计算第二维度,如果不指定,则计算张量中所有的数的均值

在这里插入图片描述2.torch.argmax()
我们想要求每一行最大的列标号,我们就要指定dim=1,表示我们不要列了,保留行的size就可以了。
假如我们想求每一列的最大行标,就可以指定dim=0,表示我们不要行了。
例题:
在这里插入图片描述

张量学习

Tensors与Numpy中的 ndarrays类似,但是在PyTorch中 Tensors 可以使用GPU进行计算.
1.torch.empty(5,3) 未初始化

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值