pytorch 鲜见操作

对两个 tensor 做逻辑运算,比如 & (与)

python 复制代码
import torch

# 定义两个张量
tensor_a = torch.tensor([1, 0, 1, 0], dtype=torch.bool)
tensor_b = torch.tensor([1, 1, 0, 0], dtype=torch.bool)

# 执行逻辑与操作
result = tensor_a & tensor_b

print(result)

结果输出:

复制代码
tensor([ True, False, False, False])

使用布尔类型张量做索引(也叫掩码)

python 复制代码
import torch

# 创建一个数据张量
data = torch.tensor([10, 20, 30, 40])

# 创建一个布尔索引张量
mask = torch.tensor([True, False, False, False])

# 使用布尔索引选择元素
selected_data = data[mask]

print(selected_data)

打印日志:

复制代码
tensor([10])

相关推荐
ACP广源盛139246256731 小时前
(ACP广源盛)GSV1175---- MIPI/LVDS 转 Type-C/DisplayPort 1.2 转换器产品说明及功能分享
人工智能·音视频
胡耀超1 小时前
隐私计算技术全景:从联邦学习到可信执行环境的实战指南—数据安全——隐私计算 联邦学习 多方安全计算 可信执行环境 差分隐私
人工智能·安全·数据安全·tee·联邦学习·差分隐私·隐私计算
Turnsole_y3 小时前
pycharm自动化测试初始化
python·selenium
停停的茶3 小时前
深度学习(目标检测)
人工智能·深度学习·目标检测
Y200309163 小时前
基于 CIFAR10 数据集的卷积神经网络(CNN)模型训练与集成学习
人工智能·cnn·集成学习
老兵发新帖3 小时前
主流神经网络快速应用指南
人工智能·深度学习·神经网络
weixin-a153003083164 小时前
[数据抓取-1]beautifulsoup
开发语言·python·beautifulsoup
AI量化投资实验室4 小时前
15年122倍,年化43.58%,回撤才20%,Optuna机器学习多目标调参backtrader,附python代码
人工智能·python·机器学习
java_logo4 小时前
vllm-openai Docker 部署手册
运维·人工智能·docker·ai·容器
倔强青铜三4 小时前
苦练Python第67天:光速读取任意行,linecache模块解锁文件处理新姿势
人工智能·python·面试