跨多个维度的Pyrotch argmax我有一个4D张量,我想得到最后二维的argmaxtorch.argmax只接受整数作为“dim”参数,而不接受元组 我怎样才能做到这一点 这是我的想法,但我不知道如何匹配我的两个“指数”张量的维数or ...2024-10-03 已阅读: n次