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

相关推荐
zhangfeng11338 分钟前
把“距离过近”的节点(或端点)合并成一个,避免重复。机器学习 python
人工智能·python·机器学习
AscentStream24 分钟前
技术文档 | 使用 Pulsar Functions 构建实时 AI Pipeline
人工智能
POLOAPI25 分钟前
为什么Claude Code让传统IDE开发者"失业"?深度解析AI编程的技术革命
人工智能·ai编程·claude
沐森26 分钟前
屏幕截图 (OCR今日ai下必备工具)
人工智能
YBCarry_段松啓32 分钟前
DeerFlow:深度研究的多智能体框架
人工智能·开源·llm
玄明Hanko32 分钟前
不想写测试脚本AI帮你搞定自动化测试
人工智能·测试
lingling0091 小时前
光伏清洗机器人是什么?艾利特协作机器人如何重塑新能源运维效率
大数据·运维·人工智能
白应穷奇1 小时前
编写高性能数据处理代码 02
后端·python
用户11906708018321 小时前
掌控python的属性,描述符钩子函数
python
阿赵3D1 小时前
selenium自动化收集资料
python·selenium·测试工具·自动化