torch tensor用法

1.x = torch.Tensor(5):zero()

Tensor转int：

int(x1.cpu().data)

range与arange区别：

if __name__ == '__main__':
    import torch

    x = torch.arange(0, 6)
    print(x.size())
    print(x.type())
    y = torch.range(0, 6)
    print(y.size())
    print(y.type())

2.conf_mask = torch.ones(nB, nA, dim, dim)

conf_mask[...]=0

除法操作：

float((torch.sum(preds == labels.data)).cpu().data)/2

Tensor构造的三种方式

下面2种比较常用。

-- 最多到4维度
x = torch.Ten