1. 首页
  2. 文章中心
  3. pytorch知识宝典

torch 列最大值筛选满足条件的行或列（使用与或）

269 阅读 0 评论
我是靠谱客的博主 秀丽红牛，这篇文章主要介绍torch 列最大值筛选满足条件的行或列（使用与或），现在分享给大家，希望可以做个参考。

 

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内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
点赞(178)

相关文章

Pytorch: tensor.expand_as()
Pytorch: tensor.expand_as()
多GPU训练的模型，转换加载到单GPU测试或预训练多GPU训练的模型，转换加载到单GPU测试或预训练报错：RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cuda.device_count() is 1.解决方案：
多GPU训练的模型，转换加载到单GPU测试或预训练多GPU训练的模型，转换加载到单GPU测试或预训练报错：RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cuda.device_count() is 1.解决方案：
pytorch 指定卡1_【科研之路】pytorch 张量
pytorch 指定卡1_【科研之路】pytorch 张量
pytorch数据导入以及预处理
pytorch数据导入以及预处理
torch 列最大值筛选满足条件的行或列（使用与或）
torch 列最大值筛选满足条件的行或列（使用与或）
[GCN] 代码解析 of GitHub：Semi-supervised classification with graph convolutional networks
[GCN] 代码解析 of GitHub：Semi-supervised classification with graph convolutional networks
torch.max用法（指定维度dim）
torch.max用法（指定维度dim）
Pytorch的torch.max()使用讲解
Pytorch的torch.max()使用讲解

评论列表共有 0 条评论

立即
投稿
返回
顶部