在开始介绍argmax ()，max )之前，请如上图所示明确“行轴”和“列轴”的方向。 图中“行轴”索引范围为0-2、轴长为3、“列轴”索引范围为0-3、轴长为4。
对明理画笔的操作importtorcht=torch.tensor ([ 3，4，1，0 ]，[ 12，1，2，3 ]，[ 52，34，88， 11 ) ) t#输出是tensor ) )即11 () ) ) ) ) ) ) ) ) ) ) ) ) )0)相当于将文理的刷子进行flatten并取出最大值的#输出(values=tensor (52、34、88、11 )、indices=tensor、2、2 () ) ) ) ) ) ) ) ) ) ) ) ) (values=tensor (4，12，88 )，indices=tensor ) (1，02() ) ) ) ) ) ) ) ) ) )。) 11 ) ) t.max ) dim=0) )1) #输出与每个行轴的最大值对应的索引： tensor ) ) 2、2、 2-------------------- -但是，最大值索引#的输出是在tensor(10 ) # flatten之后与最大值88对应的索引唯一不同的是不返回索引值importnumpyasNPT=NP.array ) )的[ 52、34、88、11 ] (t.max (axis=0) )输出是阵列(52、52、54、88、11 )