文档首页 > > API参考> TBE API参考> compute接口>

te.lang.cce.compute_five2four(input, raw_shape_4D)

te.lang.cce.compute_five2four(input, raw_shape_4D)

分享
更新时间:2020/05/12 GMT+08:00

把给定5-D “NC1HWC0”数据格式转换为4-D “NCHW”数据格式。支持的数据类型:float16。

该接口在dim_conv.py中定义。

参数说明

  • input:输入tensor,5-D格式(N, C1, H, W, C0),tvm.tensor类型。
  • raw_shape_4D:转换后tensor的维度。

返回值

res_tensor:转换为4-D格式(N, C, H, W)后的tensor,tvm.tensor类型。

调用示例

import tvm
import te.lang.cce
raw_shape = (2, 32, 16, 128)
input_shape = (2,(32+15)//16,16,128,16)
in_dtype = "float16"
input = tvm.placeholder(input_shape, name='input', dtype=in_dtype)
res = te.lang.cce.compute_five2four(input, raw_shape)
# res.shape = (2, 32, 16, 128)
分享:

    相关文档

    相关产品

文档是否有解决您的问题?

提交成功!非常感谢您的反馈,我们会继续努力做到更好!
反馈提交失败,请稍后再试!

*必选

请至少选择或填写一项反馈信息

字符长度不能超过200

提交反馈 取消

如您有其它疑问,您也可以通过华为云社区论坛频道来与我们联系探讨

智能客服提问云社区提问