zoukankan      html  css  js  c++  java
  • 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,可用于一个数和数组比较,用最大或者最小值替换数组对应位置的值

  • 相关阅读:
    CF733F
    P4826
    洛谷P2687 & P1108
    CF42A
    洛谷P1858
    CF1428C
    洛谷P4981
    树形DP
    背包六讲(也不知道为啥就是六个 $QwQ$)
    2020
  • 原文地址:https://www.cnblogs.com/lzping/p/12362474.html
Copyright © 2011-2022 走看看