eniops库中reduce函数使用方法

reduce 是 eniops 中的一个常用函数,用于对张量进行降维操作。它允许你通过指定维度名称和操作类型(如求和、均值等)来简化张量的形状。

python 复制代码
import eniops
import torch

# 创建一个示例张量
x = torch.randn(2, 3, 4)

# 使用 reduce 进行降维操作
result = eniops.reduce(x, 'b c h -> b h', 'mean')

print(result.shape)  # 输出: torch.Size([2, 4])

输入张量 x 的形状为 (2, 3, 4),对应模式 'b c h'。

reduce 操作将 c 维度通过 'mean' 操作降维,最终输出形状为 (2, 4),对应模式 'b h'。

除了mean,还有sum,max等降维方式.

如下,

python 复制代码
result = eniops.reduce(x, 'b c h -> b h', 'sum')
print(result.shape)  # 输出: torch.Size([2, 4])
python 复制代码
result = eniops.reduce(x, 'b c h -> b h', 'max')
print(result.shape)  # 输出: torch.Size([2, 4])
相关推荐
Kingairy几秒前
Pytest 插件:pytest_runtest_protocol
python·pytest
sheji34166 分钟前
【开题答辩全过程】以 基于python爬虫对微博数据可视化及实现为例,包含答辩的问题和答案
爬虫·python·信息可视化
Moonbit9 分钟前
MoonBit Pearls Vol.08: MoonBit 与 Python集成指南
后端·python·程序员
CodeDevMaster25 分钟前
从零开始:用uv构建并发布一个Python CLI应用,集成CI/CD自动化发布与Docker容器化部署
python·ci/cd·docker
二闹28 分钟前
三招搞定Python定时任务,总有一款适合你
后端·python
文人sec30 分钟前
性能测试-jmeter7-元件提取器
python·jmeter·prometheus·模块测试
这里有鱼汤40 分钟前
量化小白必看|MiniQMT踩坑记:想做实盘这些知识请你一定要掌握
后端·python
数据小子21442 分钟前
【自记】Python 中 nonlocal 和 global 的区别,以及闭包的作用和示例
python
WSSWWWSSW1 小时前
Python OpenCV图像处理与深度学习:Python OpenCV视频处理入门
图像处理·python·opencv
胖墩会武术1 小时前
【PyTorch项目实战】SAM(Segment Anything Model) —— 致力于建立第一个图像分割基础模型
人工智能·pytorch·python·sam