一 Pytorch基本语法

12 篇文章 5 订阅 ¥9.90 ¥99.00
本文介绍了Pytorch的基本语法,包括张量的概念、创建与操作,以及与Numpy数组之间的转换。Pytorch是一个支持GPU加速的科学计算包,同时也是一种深度学习平台。文章详细阐述了张量的初始化、形状改变、加法运算以及如何在张量和Numpy数组间转换,强调了in-place操作和共享内存空间的特点。
摘要由CSDN通过智能技术生成

目录

1 什么是pytorch

2 基本元素操作

3 基本运算操作

4 关于torch tensor 与 numpy array 之间的转换


1 什么是pytorch

Pytorch是一个基于Numpy的科学计算包,向他的使用者提供了两大功能

  • 作为Numpy的替代者,向用户提供使用GPU强大功能的能力
  • 作为一款深度学习的平台,向用户提供更大的活性与速度

2 基本元素操作

  • Tensors张量:张量的概念类似于Numpy的ndarray数据结构,最大的区别就是张量可以利用GPU的加速功能
from __future__ import print_function
import torch
  • 创建矩阵的操作

  - 创建一个没有初始化的矩阵

x = torch.empty(5, 3)
print(x)

  - 输出结果

tensor([[7.9898e-39, 1.1112e-38, 8.9082e-39],
        [9.1837e-39, 1.0928e-38, 8.4490e-39],
        [9.6428e-39, 1.1112e-38, 9.5511e-39],
        [1.0102e-38, 1.0286e-38, 1.0194e-38],
        [9.6429e-39, 9.2755e-39, 9.1837e-39]])

  - 创建一个有初始化的矩阵

x = torch.rand(5,4)
print(x)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
PyTorch是一个用于科学计算的开源机器学习库。以下是一些PyTorch的基础语法: 1. 创建张量: 通过torch.tensor()函数创建张量,可以指定张量的形状和数据类型。例如,torch.tensor([2.,3.2])创建了一个形状为的浮点张量。torch.Tensor()函数也可以用于创建张量,可以指定形状作为参数。例如,torch.Tensor(2,3)创建了一个形状为[2,3]的张量。默认情况下,PyTorch的张量类型是torch.FloatTensor。 2. 改变默认类型: 使用torch.set_default_tensor_type()函数可以改变默认张量类型。例如,torch.set_default_tensor_type(torch.DoubleTensor)可以将默认类型改为torch.DoubleTensor。 3. 初始化张量: 使用torch.rand()函数可以创建一个具有随机分布的张量。例如,a1=torch.rand(4,3,16,32)创建了一个形状为[4,3,16,32]的张量。 4. 张量的拼接: 使用torch.cat()函数可以将两个张量沿指定维度拼接在一起。例如,torch.cat([a1,a2], dim=2)可以将a1和a2沿第2个维度拼接在一起,得到形状为[4,3,32,32]的张量。另外,使用torch.stack()函数也可以实现拼接,不同之处在于它会额外创建一个新的维度。例如,torch.stack([a1,a2], dim=2)得到形状为[4,3,2,16,32]的张量。 5. 张量的分割: 使用torch.split()函数可以将一个张量沿指定维度分割成多个子张量。例如,torch.split(b, 2)可以将长度为2的张量b分割成两个子张量。 这些是PyTorch的一些基础语法,可以帮助你开始使用PyTorch进行科学计算和机器学习。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

OR_0295

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值