Pytorch Task1:PyTorch的基本概念

PyTorch是一个Python科学计算库,适合NumPy替代并提供深度学习平台。因其动态图特性,PyTorch易于使用和调试。本文将介绍PyTorch的安装方法和基础概念,包括张量的使用和运算操作。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

PyTorch的基本概念

什么是Pytorch,为什么选择Pytroch?

PyTorch是一个基于python的科学计算包,主要针对两类人群:

  • 作为NumPy的替代品,可以利用GPU的性能进行计算
  • 作为一个高灵活性、速度快的深度学习平台

Pytorch相比Tensorflow而言,它的设计初衷是简单易用用,所以它是基于动态图进行实现的,从而方便调试。当然,Tensorflow在1.5版的时候就引入了Eager Execution机制实现了动态图,但它还是默认使用静态图。

PyTorch的安装]

我是用按官网中用conda安装,由于没有GPU,所以用的是CPU版本

conda install pytorch-cpu torchvision-cpu -c pytorch

PyTorch基础概念

张量:

Tensor(张量)类似于NumPy的ndarray,但还可以在GPU上使用来加速计算。

from __future__ import print_function
import torch
x = torch.empty(5, 3)
print(x)
tensor([[2.2391e-19, 4.5869e-41, 1.4191e-17],
        [4.5869e-41, 0.0000e+00, 0.0000e+00],
        [0.0000e+00, 0.0000e+00, 0.0000e+00],
        [0.0000e+00, 0.0000e+00, 0.0000e+00],
        [0.0000e+00, 0.0000e+00, 0.0000e+00]])

运算

一种运算有多种语法。在下面的示例中,我们将研究加法运算。

加法:形式一

y = torch.rand(5, 3)
print(x + y)
tensor([[ 2.5541,  0.0943,  0.9835],
        [ 1.4911,  1.3117,  <
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值