zoukankan      html  css  js  c++  java
  • pytorch入门1.0(基本操作)

    pytorch入门1.0
    主要是关于张量的创建、运算、索引等一些基本操作。随便练习一下,增加对张量操作的熟悉程度。

    1.pytorch是什么?

    pytorch是2017年由Facebook人工智能研究院(FAIR)基于Torch推出的一个开源Python机器学习库。该库能借助GPU加速张量的计算;亦具有自动求导系统。

    2.使用pytorch的前期工作

    1. 首先得安装好;(假定你已经安装好啦)
    2. 导入torch包;
    import torch
    print(torch.__version__)  # 打印pytorch的版本
    
    # 设置随机种子,保证在神经网络中的各种权重的初始化一致。
    torch.manual_seed(42)
    torch.cuda.manual_seed(42)
    

    3.创建几个tensor(张量)玩玩

    # 创建空的张量(值为垃圾值)
    torch.empty(1)
    torch.empty(3).dtype
    
    # 创建张量(值为垃圾值)
    torch.Tensor(1)
    torch.Tensor(1).dtype
    
    #依照元素值创建张量,更像是类型转换
    torch.tensor([1.0,3.0])
    torch.tensor([1.0,3.0]).dtype
    
    # 创建二维矩阵,对角线元素为1,其他位置元素为0
    torch.eye(3,dtype=torch.float32)
    
    # 创建一个形状跟参数里的Tensor一样的张量
    torch.empty_like(torch.Tensor(3,3))
    
    # 创建一个全是1的向量,例如2*3*4
    torch.ones(2,3,4,device='cuda')  # 把此变量放到gpu上,devive='cuda'
    
    # 创建一个全0向量
    torch.zeros(2,4)
    
    # 创建一个3*3*3的张量,每个元素从2到5[2,5)随机取值
    torch.randint(2,5,(3,3,3))
    
  • 相关阅读:
    网摘习惯
    关于Application.DoEvents()
    五句话足以改变一生
    ActionForm中reset()的用法
    Java的MD5加密和解密类
    ibatis主键自动生成
    Parameter index out of range (3 > number of parameters
    ibatis 2.0 3.0 DTD
    SmartUpload在servlet中使用方法
    The prefix "tx" for element "tx:annotationdriven" is not bound.
  • 原文地址:https://www.cnblogs.com/datasnail/p/13082277.html
Copyright © 2011-2022 走看看