[学習] pytorch unsqueeze（）関数

unsqueeze（）関数

処置：テンソルの大きさを増します。

import torch
import numpy as np

a =  torch.tensor(np.arange(0, 6))
print(a)
print(a.shape)

a = a.unsqueeze(1)
print(a)
print(a.shape)