unsqueeze需要,否則報錯
expand(3,2,2)參數就是目标次元。
是複制方式,最後的次元必須能整除
if __name__ == '__main__':
import torch
x = torch.Tensor([[1,2], [2,3], [3,4]])
print(x.size())
print(x)
d=x.unsqueeze(1).expand(3,2, 2)
print(d)
unsqueeze需要,否則報錯
expand(3,2,2)參數就是目标次元。
是複制方式,最後的次元必須能整除
if __name__ == '__main__':
import torch
x = torch.Tensor([[1,2], [2,3], [3,4]])
print(x.size())
print(x)
d=x.unsqueeze(1).expand(3,2, 2)
print(d)