很多人会遇到需要将标签编转化成one-hot格式的矩阵,现在pytorch里面一行代码即可搞定。
import torch.nn.functional as F
one_hot_matrix = F.one_hot(y,class_number)
很多人会遇到需要将标签编转化成one-hot格式的矩阵,现在pytorch里面一行代码即可搞定。
import torch.nn.functional as F
one_hot_matrix = F.one_hot(y,class_number)