pytorch笔记：torch.nn.Threshold
Posted UQI-LIUWJ
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch笔记：torch.nn.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
'''以上是关于pytorch笔记：torch.nn.Threshold的主要内容，如果未能解决你的问题，请参考以下文章