one_hot関数【torch】
説明:
ワンホット行列を生成するための関数。
書式:
torch.nn.functional.one_hot( )
引数:
引数1:
元データとなるテンソル、torch.int64 型で構成されている必要がある。
num_classes:
ワンホット化後のベクトルのサイズ
戻り値:
ワンホット化されたテンソル