torch.argmin¶
- torch.argmin(input, dim=None, keepdim=False) LongTensor ¶
返回扁平化张量或沿某个维度的最小值(s)的索引
这是
torch.min()
返回的第二个值。有关此方法的精确语义,请参阅其文档。注意
如果存在多个最小值,则返回第一个最小值的索引。
- 参数
示例
>>> a = torch.randn(4, 4) >>> a tensor([[ 0.1139, 0.2254, -0.1381, 0.3687], [ 1.0100, -1.1975, -0.0102, -0.4732], [-0.9240, 0.1207, -0.7506, -1.0213], [ 1.7809, -1.2960, 0.9384, 0.1438]]) >>> torch.argmin(a) tensor(13) >>> torch.argmin(a, dim=1) tensor([ 2, 1, 3, 1]) >>> torch.argmin(a, dim=1, keepdim=True) tensor([[2], [1], [3], [1]])