zoukankan      html  css  js  c++  java
  • pytoch基础知识--Variable变量

     pytorch中的变量有三个属性,分别是data表示变量中的具体值, grad表示这个变量反向传播的梯度,这个的计算方式下面有专门的一个演示程序, grad_fn表示是通过什么操作得到这个变量的例如( 加减乘除、卷积、反置卷积)

    其位于torch的autograd中

    import torch
    from torch.autograd import Variable

    对比在pytorch中常用的单元tensor(张量),二者是可以相互转化的,区别在于tensor用于在CPU上计算,而Variable用于在GPU上。所以说这就是为什么pytorch加载图像的时候一般都会使用Variable变量。

    附tensor,Variable,numpy.array的转化代码

    #将numpy矩阵转换为Tensor
    sub_ts = torch.from_numpy(sub_img)   #sub_img为numpy类型
    
    #将Tensor张量转化为numpy
    sub_np1 = sub_ts.numpy()             #sub_ts为tensor张量
    
    #将numpy转换为Variable
    sub_va = Variable(torch.from_numpy(sub_img))
    
    #将Variable张量转化为numpy
    sub_np2 = sub_va.data.numpy()          #sub_va为Variable类型
    
    #将Tensor张量转化为Variable
    sub_var = Variable(sub_ts, requires_grad = True)  #sub_ts为tensor
    适当比较,砥砺前行
  • 相关阅读:
    iOS各种证书
    Error No matching provisioning profiles found
    iOS 一个开发者账号 多台Mac 共用
    外中断(学习汇编)
    端口(学习汇编)
    8086CPU寄存器
    int指令(学习汇编)
    内中断(学习汇编)
    标志寄存器(学习汇编)
    CALL和RET指令(学习汇编)
  • 原文地址:https://www.cnblogs.com/llllrj/p/15368298.html
Copyright © 2011-2022 走看看