【发布时间】：2021-12-29 11:55:28
【问题描述】：
我有一个复值二维张量 (m x n)，我想选择每一行的最小值。但torch.min() 仅支持非复杂值。我不想在元素上使用 for 循环或其他东西进行迭代。我尝试idx = torch.argmin(torch.abs(errors), dim = 1, keepdim = True) 选择张量“错误”的索引，但我不知道如何选择具有多个索引的元素。 torch.index_select() 仅支持固定索引，不支持不同行的更改。
是否有一个无需遍历每个元素的优雅解决方案？
【问题讨论】：
标签： function pytorch min torch complex-numbers