ASTHNONT

摘要： 在进行深度学习张量计算时，经常要获取张量在某个维度的最大值和最小值，以及这些值的位置。如果只需要知道位置，则torch.argmax和torch.argmin函数便可以实现。 Torch.argmax(input, dim=None, keepdim=False):返回指定维度最大值的序号。 有时候 阅读全文