Pytorch(二) Tensor Tensor的創建

Tensor是什麼

張量,是標量、向量、矩陣的高維擴展

    0維張量,標量
    1維張量,向量
    2維張量,矩陣
    n維張量


    
Variable是torch.autograd中的數據類型,主要用於封裝Tensor,進行自動求導
        
        data: 被包裝的Tensor
        grad: data的梯度
        grad_fn: 創建Tensor的Function, 是自動求導的關鍵
        requires_grad: 指示是否需要梯度
        is_leaf: 指示是否是葉子結點(張量)
        
Pytorch0.4.0版本開始,Variable併入Tensor

        這四個屬性是數據有關的屬性
            data:
            dtype: 張量的數據類型,如torch.FloatTensor, torch.cuda.FloatTensor
            shape: 張量的形狀,如(64, 3, 224, 224)
            device: 張量所在設備,GPU/CPU是加速的關鍵
            
        這個是求導有關的屬性 
            requries_grad:
            grad:
            grad_fn:
            is_leaf:

Tensor的創建

    1.直接創建
    2.依據數據創建
    3.依據概率創建

直接創建

依據數據創建

依據概率創建

發佈了41 篇原創文章 · 獲贊 5 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章