通常,对于密钥检测网络,假设原图像尺寸【550、550】,在密钥的分类为1 .输出的特征量图的尺寸为【1,550、550】的情况下,接着gt也为【1,550、550】的jdd
inputtensor([1.2.],[3.4.],dtype=torch.float64 ) targettensor ) [5.6.],dtype=torch.)。 target ) resulttensor(10 .dtype=torch.float64 ) input中的每个维和gt计算差的幂的总和,累积多个维的值,并将其除以input像素值的数量(H*W )
这样,在再推理的过程中,也可以使用特征图将多个维度累积后再除以维度;