初めてのPytorch #変換

テストデータを受け取る際に処理しやすい値に変換する処理

Lambdaを使用するためimportを追加
[0,0,0,0,0,0,0,0,0,0]を作成しyの値を1に変化させる
例：y=3なら[0,0,0,1,0,0,0,0,0,0]
これを行うことで誤差を計算しやすくしている

import torch
from torchvision import datasets
from torchvision.transforms import ToTensor, Lambda

ds = datasets.FashionMNIST(
    root="data",
    train=True,
    download=True,
    transform=ToTensor(),
    target_transform=Lambda(lambda y: torch.zeros(10, dtype=torch.float).scatter_(0, torch.tensor(y), value=1))
)