numpyで?ってなったことをまとめる。イディオム的に。
1. image.shape==[3.64,64] を [1,3,64,64]に変更する
image=np.stack([image])
2. numpyをTensor型にしてcudaに乗せる
image=torch.from_numpy(image).cuda()
3. csvから数字データを取ってきてnumpy配列に格納する
with open('name.csv','r') as f:
reader=csv.reader(f)
l=[row for row in reader]
l=np.array(l).astype('int64')
csvから取ってきた後にastypeでキャストするのを忘れないようにする