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
相关文章
相关标签/搜索