pytorch 筛选数据(使用与或)

目录

pytorch 筛选数据(使用与或)

pytorch筛选赋值


pytorch 筛选数据(使用与或)

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<

你可能感兴趣的:(pytorch知识宝典,深度学习基础,pytorch,深度学习,人工智能)