多分类不需要满足y_pred和y_gt的维度相同

二分类中,torch.nn.BECLoss 和torch,nn.BCEWithLogitsLoss需要预测输出y_pred和y_gt需要一样的维度。
故需要使用一下两种转变y_gt的方法。
假设y_pred如下:
[[1.2 2.3 ],
[2.1 2.2 ]]
其中,y_gt如下:
[[0],[1]]
目标是是转化成下面的维度:
[[1,0],
[0,1]]
代码如下:
- 直接使用list的index属
多分类不需要满足y_pred和y_gt的维度相同

二分类中,torch.nn.BECLoss 和torch,nn.BCEWithLogitsLoss需要预测输出y_pred和y_gt需要一样的维度。
故需要使用一下两种转变y_gt的方法。
假设y_pred如下:
[[1.2 2.3 ],
[2.1 2.2 ]]
其中,y_gt如下:
[[0],[1]]
目标是是转化成下面的维度:
[[1,0],
[0,1]]
代码如下:

被折叠的 条评论
为什么被折叠?