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