[ 组合替代实现 ]torch.aminmax¶

torch.aminmax(input, *, dim=None, keepdim=False, out=None)

# PyTorch 写法
y = torch.aminmax(input, dim=-1, keepdim=True)

# Paddle 写法
y = tuple([paddle.amin(input, axis=-1, keepdim=True), paddle.amax(input, axis=-1, keepdim=True)])