pytorch 獲取tensor維度信息示例
更新時間:2020年01月03日 15:47:25 作者:siqi_fighting
今天小編就為大家分享一篇pytorch 獲取tensor維度信息示例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了,直接上代碼吧!
>>> import torch >>> from torch.autograd import Variable >>> from torch import IntTensor >>> var = Variable(IntTensor([[1,0],[0,1]])) >>> var Variable containing: 1 0 0 1 [torch.IntTensor of size 2x2] >>> var.size() torch.Size([2, 2]) >>> list(var.size()) [2, 2]
以上這篇pytorch 獲取tensor維度信息示例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
回歸預測分析python數(shù)據(jù)化運營線性回歸總結(jié)
本文主要介紹了python數(shù)據(jù)化運營中的線性回歸一般應用場景,常用方法,回歸實現(xiàn),回歸評估指標,效果可視化等,并采用了回歸預測分析的數(shù)據(jù)預測方法2021-08-08
Keras存在自定義loss或layer怎樣解決load_model報錯問題
這篇文章主要介紹了Keras存在自定義loss或layer怎樣解決load_model報錯問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-09-09

