pytorch 獲取tensor維度信息示例
更新時(shí)間:2020年01月03日 15:47:25 作者:siqi_fighting
今天小編就為大家分享一篇pytorch 獲取tensor維度信息示例,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了,直接上代碼吧!
>>> 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)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
回歸預(yù)測分析python數(shù)據(jù)化運(yùn)營線性回歸總結(jié)
本文主要介紹了python數(shù)據(jù)化運(yùn)營中的線性回歸一般應(yīng)用場景,常用方法,回歸實(shí)現(xiàn),回歸評估指標(biāo),效果可視化等,并采用了回歸預(yù)測分析的數(shù)據(jù)預(yù)測方法2021-08-08
Keras存在自定義loss或layer怎樣解決load_model報(bào)錯(cuò)問題
這篇文章主要介紹了Keras存在自定義loss或layer怎樣解決load_model報(bào)錯(cuò)問題,具有很好的參考價(jià)值,希望對大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-09-09
Python使用DEAP庫實(shí)現(xiàn)簡單遺傳算法
DEAP是一個(gè)用于快速原型設(shè)計(jì)和實(shí)驗(yàn)的進(jìn)化計(jì)算框架,它支持多種進(jìn)化算法,包括遺傳算法、遺傳編程等,下面我們來看看如何使用DEAP庫實(shí)現(xiàn)簡單遺傳算法吧2025-01-01
Python實(shí)現(xiàn)人臉識別的詳細(xì)圖文教程
人臉識別是人工智能的一個(gè)重要應(yīng)用,下面這篇文章主要給大家介紹了關(guān)于Python實(shí)現(xiàn)人臉識別的相關(guān)資料,文中通過圖文介紹的非常詳細(xì),需要的朋友可以參考下2022-08-08

