PyTorch中loss.mean(dim=1)后的shape变化
经过 `loss.mean(dim=1)` 操作后，`loss` 变量的 shape 变为 `(1,)`。
原文地址: https://www.cveoy.top/t/topic/nGZk 著作权归作者所有。请勿转载和采集!
安全问答是一个知识全球问答,包含丰富的问答知识
经过 `loss.mean(dim=1)` 操作后，`loss` 变量的 shape 变为 `(1,)`。
原文地址: https://www.cveoy.top/t/topic/nGZk 著作权归作者所有。请勿转载和采集!