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
除法操作ÿ