当前位置 主页 > 网站技术 > 代码类 >

    pytorch 获取tensor维度信息示例

    栏目:代码类 时间:2020-01-03 18:08

    我就废话不多说了,直接上代码吧!

    >>> 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维度信息示例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持IIS7站长之家。