更新时间:2021-03-18 GMT+08:00
te.lang.cce.vadd(lhs, rhs)
两个tensor按元素相加,元素的数据类型需要一致,支持的类型:float16,float32,int32。int8,uint8会被转换为float16。
该接口在elewise_compute.py中定义。
参数说明
- lhs:左tensor,tvm.tensor类型。
- rhs:右tensor,tvm.tensor类型。
返回值
res_tensor:表示lhs + rhs,tvm.tensor类型
调用示例
shape = (1024,1024) input_dtype = "float16" data1 = tvm.placeholder(shape, name="data1", dtype=input_dtype) data2 = tvm.placeholder(shape, name="data2", dtype=input_dtype) res = te.lang.cce.vadd(data1, data2)
父主题: compute接口