0
点赞
收藏
分享

微信扫一扫

pytorch中的数据集处理部分data_transforms = { ‘train‘: transforms.Compose([...])...

juneyale 2024-05-07 阅读 12

一 基本语法

1 torch中tensor的声明

声明的时候有的时候需要指出数据的类型,不然在kernel中数据类型无法匹配

2 idx

idx表示的是数据在sram中的索引,如果idx为(),则表示为只有一个数据

二 triton中函数

1 sum

如果输入是torch中的声明的话,则输出为

 这就是一个reduce的过程,将x轴的数据全部相加,得到一个数字

2 乘法

2.1 矩阵乘

dot

2.2 卷积乘(叉乘)

2.3 具体说明

举报

相关推荐

0 条评论