pytorch中max函数 -preb.max(-1)[1]

https://blog.csdn.net/jzwong/article/details/103267516#commentBox

1.pytorch中二维向量（Tensor）都是行向量，列堆叠。

结果示例：

2.

 （-1）是维度，（-1）是从右往左第一个维度，也就是说最后一个维度，在二维向量中最后一个维度是1，也就是按列堆叠，也就是沿行取最大值。

结果示例：

3.

 按照行（-1维）取出最大值的索引。

结果示例：

 

4.

按照行（-1维）取出最大值，以及最大值的索引。

结果示例：