1 基本用法

torch.clamp(input, min=None, max=None, *, out=None)

使得tensor中比min小的變成min，比max大的變成max

2 使用舉例

import torch a = torch.randn(4) print(a) #tensor([-1.7120, 0.1734, -0.0478, -0.0922])torch.clamp(a, min=-0.5, max=0.5) #tensor([-0.5000, 0.1734, -0.0478, -0.0922])

總結(jié)

以上是生活随笔為你收集整理的pytorch笔记 torch.clamp（截取上下限）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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