以二维numpy矩阵为例
import torch
import numpy as np
K=3 #取每行最小3个值的索引
data=np.random.rand(4,7)
print(data)
data=torch.from_numpy(data)
a, idx = torch.sort(data, descending=False)
lists=idx[:,:K]
print(lists)
运行结果如下：
results.jpg
import torch
import numpy as np
K=3 #取每行最小3个值的索引
data=np.random.rand(4,7)
print(data)
data=torch.from_numpy(data)
a, idx = torch.sort(data, descending=False)
lists=idx[:,:K]
print(lists)
之前写了一个用pytorch计算二维矩阵每行最小N个值的索引的代码，示例用了一个[4,7]维的数据，在示范的时候也...
以二维numpy矩阵为例 运行结果如下：
描述 写出一个高效的算法来搜索 m × n矩阵中的值。 这个矩阵具有以下特性： 每行中的整数从左到右是排序的。每行...
写出一个高效的算法来搜索 m × n矩阵中的值。 这个矩阵具有以下特性： 每行中的整数从左到右是排序的。每行的第一...
描述 给定一个5*5的矩阵，每行只有一个最大值，每列只有一个最小值，寻找这个矩阵的鞍点。鞍点指的是矩阵中的一个元素...
写出一个高效的算法来搜索 m × n矩阵中的值。这个矩阵具有以下特性：每行中的整数从左到右是排序的。每行的第一个数...
写出一个高效的算法来搜索 m × n矩阵中的值。这个矩阵具有以下特性：每行中的整数从左到右是排序的。每行的第一个数...
知识点 给定一个5*5的矩阵，每行只有一个最大值，每列只有一个最小值，寻找这个矩阵的鞍点。鞍点指的是矩阵中的一个元...
【题目描述】写出一个高效的算法来搜索m×n矩阵中的值，返回这个值出现的次数。 这个矩阵具有以下特性： 每行中的整数...
写出一个高效的算法来搜索m×n矩阵中的值，返回这个值出现的次数。这个矩阵具有以下特性：每行中的整数从左到右是排序的...
本文标题：pytorch 计算2维矩阵每行最小N个值的索引
本文链接：https://www.haomeiwen.com/subject/bqunhrtx.html
网友评论