入门pytorch

什么是pytorch?

PyTorch即 Torch 的 Python 版本。Torch 是由 Facebook 发布的深度学习框架,因支持动态定义计算图,相比于 Tensorflow 使用起来更为灵活方便,特别适合中小型机器学习项目和深度学习初学者。但因为 Torch 的开发语言是Lua,导致它在国内一直很小众。所以,在千呼万唤下,PyTorch应运而生!PyTorch 继承了 Troch 的灵活特性,又使用广为流行的 Python 作为开发语言,所以一经推出就广受欢迎!

PyTorch是基于python的科学计算包,与numpy类似,但又能利用GPU来加速计算。

  • Tensors
    创建一个5×3未初始化的矩阵
import torch
x = torch.Tensor(5,3)
print (x)

结果:

1.00000e-37 *
  8.4745  0.0000  0.0000
  0.0000  0.0000  0.0000
  0.0000  0.0000  0.0000
  0.0000  0.0000  0.0000
  0.0000  0.0000  0.0000
[torch.FloatTensor of size 5x3]

创建一个5×3随机初始化的矩阵

import torch

x = torch.rand(5,3)
print (x)
--------------------------------
0.8031  0.4576  0.2146
 0.0219  0.8962  0.9953
 0.6851  0.8307  0.0741
 0.4235  0.4125  0.7773
 0.4069  0.5352  0.9514
[torch.FloatTensor of size 5x3]

获取其大小size
print (x.size())
输出torch.size([5,3])

  • operations
    对于torch的运算操作有多种语法,比如来看Add操作
    1.直接相加
y = torch.rand(5, 3) 
  print(x + y)

2.使用add()加法函数
print(torch.add(x, y))
3 .给出一个输出张量

result = torch.Tensor(5, 3)
torch.add(x, y, out=result)
print(result)

4.替换(in-place)

# adds x to y
y.add_(x)
print(y)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 之前有很多小伙伴私信我说文章思想能看懂,但是pytorch的部分因为没有看过pytorch教程所以一脸懵逼。对此我...
    SherlockLiao阅读 3,462评论 1 16
  • 官方所有教程的地址:pytorch.org/tutorials 以下是基于实例来入门pytorch Learnin...
    MiracleJQ阅读 1,844评论 0 4
  • 前面一节我们讲了cnn以及如何使用pytorch实现简单的多层卷积神经网络,下面我们将进入rnn,对于rnn我也涉...
    SherlockLiao阅读 1,342评论 1 4
  • 上一节介绍了简单的线性回归,如何在pytorch里面用最小二乘来拟合一些离散的点,这一节我们将开始简单的logis...
    SherlockLiao阅读 9,623评论 0 10
  • 上一篇教程我们基本的介绍了pytorch里面的操作单元,Tensor,以及计算图中的操作单位Variable,相信...
    SherlockLiao阅读 2,516评论 2 5