torch.argmax¶. 火炬。 argmax ( 输入)→LongTensor¶. 属性中所有元素的最大值的索引 输入 张量。 这是第二个返回值 torch.max () .有关此方法的确切语义,请参阅其 ...
確定! 回上一頁