#用max得到的結(jié)果,設(shè)置keepdim為True,避免降維。因?yàn)閠opk函數(shù)返回的index不降維,shape和輸入一致。 _, indices_max=pred.max(dim=1, keepdim=True) print(indices_max==indices) # pred tensor([[-0.1480,-0.9819,-0.3364,0.7912,-0.3263], [-0.8013,-0.9083,0.7973,0.1458,-0.9156], ...
www.dbjr.com.cn/article/1777...htm 2025-5-26