pytorch max和clamp
Posted lzping
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch max和clamp相关的知识，希望对你有一定的参考价值。
torch.max()
torch.max(a)：数组a的最大值
torch.max(a, dim=1)：多维数组沿维度1方向上的最大值，若a为二维数组，则为每行的最大值(此时是对每行的每列值比较取最大，即沿列的方向最大值)
torch.max(a,b)：对同样大小的两个数组比较，取对应位置上的最大值
torch.clamp()
torch.clamp(a, min, max)：对数组a的每个元素，超过max的取max,小于min的取min,可用于一个数和数组比较，用最大或者最小值替换数组对应位置的值
以上是关于pytorch max和clamp的主要内容，如果未能解决你的问题，请参考以下文章
Pytorch Autograd 在使用 .clamp 而不是 torch.relu 时会给出不同的渐变
CSS中 min() max() clamp()函数及使用场景详解