计算PyTorch中4D张量的最大值和最小值


一瓶泡沫
2025-03-13 10:52:45 (20天前)


例如,假设我们有一个4维张量

进口火炬
X = torch.rand(2,3,4,4)……

2 条回复
  1. 0# 银环蛇 | 2019-08-31 10-32



    只需按顺序计算两个维度的最大值,它会得到相同的结果:




    1. tup = (2,3)
      for dim in tup:
      X = torch.max(X,dim=dim,keepdim=True)[0]

    2. </code>

登录 后才能参与评论