帝王谷资源网 Design By www.wdxyy.com

我就废话不多说了,大家还是直接看代码吧~

import torch
input_tensor = torch.tensor([1,2,3,4,5])
print(input_tensor>3)
mask = (input_tensor>3).nonzero()
print(mask)
print(input_tensor.index_select(0,mask))
tensor([0, 0, 0, 1, 1], dtype=torch.uint8)
tensor([3, 4])
tensor([4, 5])

补充知识:pytorch tensor筛选满足条件的行或列(使用与或)

我就废话不多说了,大家还是直接看代码吧~

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)

以上这篇使用pytorch 筛选出一定范围的值就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

标签:
pytorch,筛选,范围值

帝王谷资源网 Design By www.wdxyy.com
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
帝王谷资源网 Design By www.wdxyy.com

RTX 5090要首发 性能要翻倍!三星展示GDDR7显存

三星在GTC上展示了专为下一代游戏GPU设计的GDDR7内存。

首次推出的GDDR7内存模块密度为16GB,每个模块容量为2GB。其速度预设为32 Gbps(PAM3),但也可以降至28 Gbps,以提高产量和初始阶段的整体性能和成本效益。

据三星表示,GDDR7内存的能效将提高20%,同时工作电压仅为1.1V,低于标准的1.2V。通过采用更新的封装材料和优化的电路设计,使得在高速运行时的发热量降低,GDDR7的热阻比GDDR6降低了70%。