天天看點

torch擴充次元

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)