今天的主场:torch.sum(). 最近在尝试一个loss,其中涉及在一个4维的tensor中求后两维的和,因为在pytorch框架下训练批量的图片格式,通常都是4维格式(batchsize ...
確定! 回上一頁