ReFormat
支持的产品型号
Atlas 训练系列产品
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 推理系列产品(Ascend 310P处理器)
函数功能
该函数不改变tensor数据的值,在指定format和输入x的维度相同时,将输入数据格式设置为目标format。
具体功能是将用户输入tensor的viewFormat、originalFormat、storageFormat统一为指定的format。
函数原型
const aclTensor *ReFormat(const aclTensor *x, const op::Format &format, aclOpExecutor *executor=nullptr)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
x |
输入 |
需要被转换的tensor。数据类型支持FLOAT16、FLOAT32、INT32、UINT32、INT8、UINT8。 |
format |
输入 |
输入tensor要转换的目标format。 |
executor |
输入 |
op执行器,包含了算子计算流程。 |
返回值说明
返回数据格式为目标format后的tensor。
约束说明
输入tensor的数据的维度需要与指定format的维度相同。
调用示例
// 将输入reformat成NCHW格式 auto reformatInput = l0op::ReFormat(unsqueezedInput, Format::FORMAT_NCHW); CHECK_RET(reformatInput != nullptr, nullptr);
父主题: 基础核函数接口