pytorch中Tensor的數據類型

1. 32位浮點型:torch.FloatTensorpython a=torch.Tensor( [[2,3],[4,8],[7,9]], ) print "a:",a print "a.size():",a.size() print "a.dtype:",a.dtype b=torch.FloatTensor( [[2,3],[4,8],[7,9]] ) print "b:",b print
相關文章
相關標籤/搜索