栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

mindspore lite的MStensor有没有使得MStensor合并以及进行高维度转置的API呢

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

mindspore lite的MStensor有没有使得MStensor合并以及进行高维度转置的API呢

问题描述:

【功能模块】

mindspore lite有没有使得MStensor合并以及进行高维度转置的API呢

【问题现象】

在进行推理代码的编写的时候遇到了需要将两个图像的tensor合并并进行特定的转置,类似numpy中的transpose([1, 2, 0])

但是在mindspore lite当中没有找到特定转置相关的API,找到的和合并相关的只有mindspore::dataset::TensorTransform下的Concatenate 这个API

求助一下大佬们是否有转置相关的API?感觉这种常用接口应该有才对,可能是我一直没有找到。

解答:

我看了一下这个算子的源码实现(mindsporeccsrcminddatadatasetkernelsimagelite_cvimage_process.cc):

 

1267行到1270行是转置过程,而且可以看出来是二维的转置。src应该是转置前端矩阵,dst是转置后的矩阵(需要初始化形状)

这里有例子: https://www.mindspore.cn/lite/api/en/master/generate/function_mindspore_dataset_Transpose-1.html

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/969018.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号