小寫的tensor接受數據,大寫的Tensor()接受的是shape,數據的維度

傳入數據用tensor,傳入形狀用FloatTensor

 

1.  小寫的tensor:接受數據

2.  大寫的Tensor()或者FloatTensor():接受的是shape,數據的維度

# 2 import from list
# 傳入數據用tensor,傳入形狀用FloatTensor
# 小寫的tensor接受數據,
# 大寫的Tensor()或者FloatTensor()接受的是shape,數據的維度

b = torch.tensor([2., 3.2])
print(b)
# out:tensor([2.0000, 3.2000])
 
b = torch.FloatTensor([2., 3.2])
print(b)
# out:tensor([2.0000, 3.2000])
# 少用,容易混淆:也可以接受數據,在列表裏面,

b = torch.FloatTensor(2, 3)  # 兩行三列
print(b)
# out:tensor([[1.0790e-43, 0.0000e+00, 1.4013e-45],
#         [0.0000e+00, 1.4013e-45, 0.0000e+00]])

b = torch.tensor([[2., 3.2], [1., 22.3]])
print(b)
# tensor([[ 2.0000,  3.2000],
#         [ 1.0000, 22.3000]])

 

參考: https://blog.csdn.net/weixin_39450145/category_9627197.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章