pytorch 獲取tensor維度資訊示例
阿新 • • 發佈:2020-01-09
我就廢話不多說了,直接上程式碼吧!
>>> 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維度資訊示例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。