兩個函數(shù)用法相同，此處就介紹max函數(shù)。

1.torch.max(data),不指定維度，返回data的最大值。

2.torch.max(data,dim),返回data中指定維度的最大值。

3.torch.max(data1,data2),返回data1和data2相同位置兩值中的最大值。

?

總結(jié)

以上是生活随笔為你收集整理的pytorch 之 torch.max() 和 torch.min() 记录的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。