pytorch view、permute、linspace repeat

view:取出所有元素後,按新維度重新排序

permute:交換維度的值  ,然後重新排序

 

linspace:(start,end,分成幾份)

grid_x=torch.linspace(0,9,10)
print(grid_x)

輸出:tensor([0., 1., 2., 3., 4., 5., 6., 7., 8., 9.])

 

repeat:參數:表示該維度重複幾次

 

 

 

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