假设我们要处理图像分类问题,假设batch_size=16,num_class=10,那么进行on
e-hot编码之后的label为[16,10]的张量,而经过model后的output也应该为[16,10],假如我现在要获得正样本,负样本,应该如何用代码去实现。
目的是为了使用torch.nn.TripletMarginLoss,其要求输入正样本和负样本:
假设我们要处理图像分类问题,假设batch_size=16,num_class=10,那么进行on
e-hot编码之后的label为[16,10]的张量,而经过model后的output也应该为[16,10],假如我现在要获得正样本,负样本,应该如何用代码去实现。
目的是为了使用torch.nn.TripletMarginLoss,其要求输入正样本和负样本: