pytorch学习笔记(一)-----基础使用篇

这篇博客是PyTorch学习笔记的第一部分,主要介绍基础使用,特别是CUDA张量。PyTorch是一个用于科学计算的Python包,旨在支持深度学习研究,并能利用GPU加速计算。内容包括张量的创建、维度查看、加法操作及其在原地(in-place)修改的方法,以及如何使用numpy索引和调整张量维度。还提到了张量与numpy之间的相互转化,并介绍了将张量移动到CUDA设备上的方法。
摘要由CSDN通过智能技术生成

这是学习pytorch的学习笔记,个人记录篇

pytorch是一个基于Python的科学计算包,目标用户有两类

  • 为了使用GPU来替代numpy
  • 一个深度学习研究平台:提供最大的灵活性和速度

张量Tensor类似于numpy,但是可以使用GPU来进行相关的计算。

创建数组:
x = torch.Tensor(5, 3) 维度
x = torch.zeros(5, 3, dtype=torch.long) 维度 类型
x = torch.tensor([5.5, 3]) 内容
x = x.new_ones(5, 3, dtype=torch.double)
x = torch.randn_like(x, dtype=torch.float)

查看尺寸:
print(x.size())

执行加法;
1.y = torch.rand(5, 3)
print(x + y)
2.print(torch.add(x, y))
3.result = torch.empty(5, 3)
torch.add(x, y, out=result)
4.y.add_(x)

** 注意**
任何在原地(in-place)改变张量的操作都有一个_后缀。例如x.copy_(y), x.t_()操作将改变x.

你可以使用所有的numpy索引操作。
print(x[:, 1])

调整维度
x = torch.randn(4, 4)
y = x.view(16)
z = x.view(-1, 8) # -1的意思是没有指定维度
print(x.size(), y.size(), z.size())

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值