CANN算子库Reshape接口文档
Reshape【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs产品支持情况Ascend 950PR/Ascend 950DT 不支持Atlas A3 训练系列产品/Atlas A3 推理系列产品 支持Atlas A2 训练系列产品/Atlas A2 推理系列产品 支持Atlas 200I/500 A2 推理产品 支持Atlas 推理系列产品 支持Atlas 训练系列产品 支持功能说明该函数不改变算子tensor数据只是将用户传入的输入tensor x的shape转换成该函数的第二个参数shape。函数原型const aclTensor *Reshape(const aclTensor *x, const op::Shape shape, aclOpExecutor *executor)const aclTensor *Reshape(const aclTensor *x, const aclIntArray *shape, aclOpExecutor *executor)参数说明参数输入/输出说明x输入待转换的输入tensor。数据类型和数据格式不限制。输入必须保证是连续内存数据。shape输入转换后的目标shape支持aclIntArray*、op::Shape即gert::Shape类型。数据类型和数据格式不限制。executor输入op执行器包含了算子计算流程。返回值说明若Reshape转换成功则返回带有目标shape信息的aclTensor给调用者若失败则返回nullptr。约束说明Reshape转换成功的前提是x的ShapeSize需要和第二个参数shape的ShapeSize相等所谓的ShapeSize举例如下A的shape(1, 3, 256, 256)则A的ShapeSize1*3*256*256。当前不支持转换成空tensor所谓的空tensor即shape中包含0。调用示例void Func(const aclTensor *x, const op::Shape shape, aclOpExecutor *executor) { auto ret l0op::Reshape(x, shape, executor); return; }【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考