torch.argmax()
1. 官方文档:torch.argmax(input, dim, keepdim=False) → LongTensorReturns the indices of the maximum values of a tensor across a dimension.Parametersinput (Tensor) – the input tensor.dim (int) – the dimension to reduce. If None, the argmax of the flatt
原创
2020-07-27 23:40:37 ·
21068 阅读 ·
2 评论