pytorch max和clamp


torch.max()

torch.max(a):數組a的最大值
torch.max(a, dim=1):多維數組沿維度1方向上的最大值,若a為二維數組,則為每行的最大值(此時是對每行的每列值比較取最大,即沿列的方向最大值)
torch.max(a,b):對同樣大小的兩個數組比較,取對應位置上的最大值

torch.clamp()

torch.clamp(a, min, max):對數組a的每個元素,超過max的取max,小於min的取min,可用於一個數和數組比較,用最大或者最小值替換數組對應位置的值


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM