label_one_hot = torch.nn.functional.one_hot(labels, self.num_classes).float().to(self.device)

https://pytorch.org/docs/stable/nn.functional.html#torch.nn.functional.one_hot

總結(jié)

以上是生活随笔為你收集整理的pytorch将label转为one hot形式的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。