美文网首页
入门pytorch

入门pytorch

作者: showaichuan | 来源:发表于2017-08-12 09:34 被阅读0次

    什么是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

          本文链接:https://www.haomeiwen.com/subject/wrpdrxtx.html