torch.max() / torch.min(). 對torch.max() 進行backward() 時,對輸入的tensor 中數值最大的元素的gradient 會貢獻1,其餘則為0。
確定! 回上一頁