概述
max_v= torch.max(scores_s, 1)[0]
下面的代码是一样的：
index_num = torch.arange(0, conf.size(0))
max_ids= torch.max(scores_s, 1)[1]
scores_v=scores_s[index_num,max_ids]
筛选满足条件的行或列（使用与或）
import torch
x = torch.linspace(1, 8, steps=8).view(4, 2)
print(x)
area1=(x[:,0]>5.5)&(x[:,1]>5.5)
c=x[:,0]*x[:,1]
area2=c>25
area=area1|area2
print(x[area])
if 0:
# index=torch.max(area,1)[0]
b=x[area]
# b= x[torch.where((x[:,0]>0) & (x[:,0]<6))]
# print(b)
最后
以上就是秀丽红牛为你收集整理的torch 列最大值筛选满足条件的行或列（使用与或）的全部内容，希望文章能够帮你解决torch 列最大值筛选满足条件的行或列（使用与或）所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![[GCN] 代码解析 of GitHub：Semi-supervised classification with graph convolutional networks](/uploads/reation/bcimg8.png)
发表评论 取消回复