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])

相关推荐
今***b11 分钟前
Python 操作 PPT 文件:从新手到高手的实战指南
java·python·powerpoint
迪菲赫尔曼1 小时前
大模型入门实战 | 基于 YOLO 数据集微调 Qwen2.5-VL-3B-Instruct 的目标检测任务
人工智能·yolo·目标检测·大模型·微调·新手入门·qwen2.5
MARS_AI_1 小时前
云蝠智能 Voice Agent:多语言交互时代的AI智能语音呼叫
人工智能·自然语言处理·交互·语音识别
程序员杰哥1 小时前
Jmeter+Jenkins接口压力测试持续集成
自动化测试·软件测试·python·测试工具·jmeter·jenkins·压力测试
THMAIL1 小时前
深度剖析Spring AI源码(七):化繁为简,Spring Boot自动配置的实现之秘
人工智能·spring boot·spring
Java开发-楠木1 小时前
【猿人学】web第一届 第13题 入门级 cookie
爬虫·python
司徒轩宇2 小时前
Python secrets模块:安全随机数生成的最佳实践
运维·python·安全
机器之心3 小时前
谷歌nano banana正式上线:单图成本不到3毛钱,比OpenAI便宜95%
人工智能·openai
兰亭妙微3 小时前
从线到机:AI 与多模态交互如何重塑 B 端与 App 界面设计
人工智能·小程序·交互·用户体验设计公司