例如,假设我们有一个4维张量
进口火炬X = torch.rand(2,3,4,4)……
只需按顺序计算两个维度的最大值,它会得到相同的结果:
tup = (2,3)for dim in tup: X = torch.max(X,dim=dim,keepdim=True)[0]</code>
tup = (2,3)for dim in tup: X = torch.max(X,dim=dim,keepdim=True)[0]
</code>