lua – Torch张量等效函数到matlab的“find”？

发布时间：2020-12-15 00:15:51 所属栏目：大数据来源：网络整理

导读：简而言之,我想知道火炬中是否存在张量命令,它给出了张量满足某个标准的元素索引. 这是matlab代码,说明了我希望能够在火炬中做什么： my_mat = magic(3); % returns a 3 by 3 matrix with the numbers 1 through 9greater_than_fives = find(my_mat 5); % fin

简而言之,我想知道火炬中是否存在张量命令,它给出了张量满足某个标准的元素索引.

这是matlab代码,说明了我希望能够在火炬中做什么：

my_mat = magic(3); % returns a 3 by 3 matrix with the numbers 1 through 9
greater_than_fives = find(my_mat > 5); % find indices of all values greater than 5,the " > 5" is a logical elementwise operator that returns a matrix of all 0's and 1's and finally the "find" command picks out the indices with a "1" in them
my_mat(greater_than_fives) = 0;  % set all values greater than 5 equal to 0

我知道我可以使用for循环在火炬中执行此操作,但是有一些等同于matlab的find命令可以让我更紧凑地执行此操作吗？

解决方法

x[x:gt(5)] = 0

通常有x：gt：lt：ge：le：eq

还有一般：apply函数tha接收匿名函数并将其应用于每个元素.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!