这个屌丝很懒，什么也没留下！

pytorch笔记：torch.nn.Threshold_torch threshold

作者：小舞很执着 | 2024-08-20 12:22:55

torch threshold

1 使用方法


torch.nn.Threshold(
    threshold, 
    value, 
    inplace=False)

对输入进Threshold的Tensor进行阈值操作

2 使用举例


import torch
 
a=torch.randn(2)
print((a)
#tensor([-0.6056,  1.5696])
 
m = torch.nn.Threshold(0.1, 20)
m(a)
#tensor([20.0000,  1.5696])
'''
第一项比0.1小，所以取value值20
第二项比0.1大，所以取自己的值1.5696
'''

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小舞很执着/article/detail/1007168