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]

參考https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int

以上這篇pytorch 獲取tensor維度信息示例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

python 時間處理之月份加減問題
這篇文章主要介紹了python 時間處理之月份加減問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2022-11-11
使用django自帶的user做外鍵的方法
這篇文章主要介紹了使用django自帶的user做外鍵的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2020-11-11
Python機器學習之手寫KNN算法預測城市空氣質量
KNN（K-Nearest Neighbor）最鄰近分類算法是數(shù)據(jù)挖掘分類（classification）技術中常用算法之一，本文將介紹如何通過KNN算法實現(xiàn)城市空氣質量的預測，感興趣的同學可以了解一下
2021-12-12
Python編程之黑板上排列組合，你舍得解開嗎
這篇文章主要介紹了Python排列組合算法的相關內容，具有一定參考價值，需要的朋友可以了解下。
2017-10-10
Anaconda配置各版本Pytorch的實現(xiàn)
本文是整理目前全版本pytorch深度學習環(huán)境配置指令，以下指令適用Windows操作系統(tǒng)，在Anaconda Prompt中運行，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2021-08-08
pandas數(shù)據(jù)類型之Series的具體使用
本文主要介紹了pandas數(shù)據(jù)類型之Series的具體使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2022-08-08
python人工智能TensorFlow自定義層及模型保存
這篇文章主要為大家介紹了python人工智能TensorFlow自定義層及模型保存示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步
2021-11-11
Selenium結合BeautifulSoup4編寫簡單的python爬蟲
這篇文章主要介紹了Selenium結合BeautifulSoup4編寫簡單的python爬蟲，幫助大家更好的理解和學習python 爬蟲的相關知識，感興趣的朋友可以了解下
2020-11-11
pytorch?tensor按廣播賦值scatter_函數(shù)的用法
這篇文章主要介紹了pytorch?tensor按廣播賦值scatter_函數(shù)的用法，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2023-06-06
Python分布式異步任務框架Celery使用教程
Celery 是由Python 編寫的簡單，靈活，可靠的用來處理大量信息的分布式系統(tǒng),它同時提供操作和維護分布式系統(tǒng)所需的工具
2023-05-05