Now, you can do this. The PR was merged (Aug 28) and it is now available in the nightly release. Simply use torch.amax() :
確定! 回上一頁