如果有两个Tensor

一个是A (1,3) float16

[[1,2,3]]

一个是B (3,1)float16

[[3]

[4]

[5]]

现在要设计算子 A*B 如下算子

官方的计算案例

使用 MUL来构建COMPUTE

PS :  Mul(yLocal,ALocalLocal,BLocalLocal,this->tileLength);

本来期望算子应该是输出[4,8,12,5,10,15,6,12,18]

但是输出是[4,10,18，0，0，0，0，0，0] 

非单行或者单列用mul 就没问题

如何解决这个问题？

是要在算子里针对对行单列计算进行扩充吗

比如将B 扩充成

[[3 3 3]

[4 4 4]

[5 5 5]]