view_as()函數
函數定義:view_as(tensor) [參數為一個Tensor張量]
該函數的作用是將調用函數的變量,轉變為同參數tensor同樣的形狀。
例子
data1 = [[[1, 2], [3, 4], [5, 6]], [[7, 8], [9, 0], [10, 11]]]
t1 = torch.Tensor(data1).long() # size=2,3,2
data2 = [[[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 0], [10, 11]]]
t2 = torch.Tensor(data2).long()
print(t1.size())
print(t2.size())
print("-------view_as()-------")
t2=t2.view_as(t1)
print(t2)
print(t2.size())
輸出結果:
可以看出經過view_as()操作后,t2 Tensor轉變為了與t1 相同的形狀。(需要重新對t2賦值,這是因為不是進行的原地操作