LLVM MLIR中的tensor类型怎么表示张量形状

胖萱小哥_1935

胖萱小哥_1935

2026-10-07

568人浏览

原创

mlir中tensor类型形状必须用小写x连接维度、无空格逗号,动态维用*仅限外层,标量为tensor无括号,reshape后需显式cast更新类型。

llvm mlir中的tensor类型怎么表示张量形状

张量形状直接写在 tensor 类型后,用 和 <code>> 包裹,维度间用 x 连接,不能用逗号或空格

tensor 是标准写法,不是 tensor

MLIR 的 tensor 类型中,形状与元素类型紧贴,中间无空格,维度分隔符必须是小写字母 x(不是 *、, 或 ×):

  • tensor ✅ 正确:2 行 × 3 列的 f64 张量
  • tensor ❌ 解析失败:逗号不被接受
  • tensor ❌ 错误符号:Unicode 乘号 × 不合法
  • tensor ❌ 空格破坏语法:类型字符串中不允许空格

动态维度用 * 表示,但仅限最外层或特定位置

* 表示运行时已知、编译期未知的维度大小,但它有约束:

  • tensor ✅ 合法:一维动态张量
  • tensor ✅ 合法:第一维动态、第二维固定为 3
  • tensor ✅ 合法(Toy dialect 支持),但某些 dialect(如 Linalg)可能要求动态维只能在最前
  • tensor ❌ 多数 dialect 拒绝:连续多个 * 通常不被支持

注意:* 不代表“任意维”,它只表示“该位置维度大小未定”,且 shape 推导仍需满足操作语义(例如 matmul 要求最后两维可收缩)。

Vega AI
Vega AI

Vega AI是一款由右脑科技推出的在线 AI 绘画和视频创作平台。

下载

rank-0 张量(标量)写作 tensor,不带 和数字

零维张量即标量,在 MLIR 中直接省略形状括号:

  • tensor<f64></f64> ✅ 正确:标量
  • tensorf64> ❌ 语法错误:空尖括号非法
  • tensor ❌ 不等价:这是 1 元素一维张量,不是标量;二者在语义和 lowering 中行为不同(如内存布局、广播规则)

Toy dialect 明确区分 tensor<f64></f64>(标量)和 tensor(向量),混用会导致类型检查失败或运行时越界。

形状推导失败常见于 reshape 或 transpose 后未显式标注

MLIR 不自动传播形状——即使你用 toy.reshape 把 tensor 变成 2x3,结果类型仍是 tensor,除非你显式 cast 或用 dialect 支持的 shape-refinement pass:

  • Toy dialect 中,var b = [1,2,3,4,5,6]; 会生成 tensor,因为类型声明强制了 shape
  • 但 %b = toy.reshape(%a) : tensor to tensor 之后,若下游操作(如 matmul)需要确切 shape,就会报错 “operand type mismatch”
  • 解决方法:加 tensor.cast 或用 linalg.tensor_reshape(它要求目标 shape 作为属性传入)

最容易被忽略的是:tensor 类型的 shape 是类型系统一部分,不是 metadata;改 shape 必须改类型,不能只改值。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

2026.09.30

120

10

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

2026.09.30

100

14

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

2026.09.30

80

12

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

2026.09.30

60

26

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

2026.09.29

80

15

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

2026.09.23

280

15

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

2026.09.23

180

15

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

2026.09.23

140

15

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

2026.09.22

80

12

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程