无标题无名氏No.62584427 返回主串
2024-05-31(五)09:45:10 ID:KcuTIDY 回应
用torch训练,在计算损失的时候,如果预测值和实际值是多维的,比如(16,3),那torch的criterion是怎么计算损失的,挨个算然后求均值吗( ゚∀。)?
无标题无名氏No.62587777
2024-05-31(五)15:58:03 ID: KcuTIDY (PO主)
损失函数就用的mse,是直接算均值的就行,我还想的手写个循环来着(; ゚ 3゚)