torch.clamp()函数
作者:互联网
torch.clamp(input, min, max, out=None)
将输入input张量每个元素的范围限制到区间 [min,max],返回结果到一个新张量。
input (Tensor) – 输入张量
min (Number) – 限制范围下限
max (Number) – 限制范围上限
out (Tensor, optional) – 输出张量
boxes_nms=torch.randint(-100,1000,(3,4))
print(boxes_nms)
标签:clamp,函数,min,max,torch,张量,input 来源: https://www.cnblogs.com/leifzhang/p/15408149.html