import torch
import numpy
x = torch.tensor(2,2,2)
t = x.view(1,8)#输出数组的大小为1*8
t = x.view(-1,4)#输出数组的大小为2*4, -1表示自行判断
t = x.view(8)#输出数组的大小为8*1
t = x.view(-1)#输出数组的大小为1*8torch中view函数的用法
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
import torch
import numpy
x = torch.tensor(2,2,2)
t = x.view(1,8)#输出数组的大小为1*8
t = x.view(-1,4)#输出数组的大小为2*4, -1表示自行判断
t = x.view(8)#输出数组的大小为8*1
t = x.view(-1)#输出数组的大小为1*8