ITADN

How to multipy a large tensor with smaller tensor

#14Closedbrappier 创建于 2023-06-14
B
brappiercommented
I have ``` a = Input() b = Input() c = a * b ``` After building the model, i pass a -> NCHW((2,3,4,5) b -> .N(1) i get a shape mismatch error Basically i want to build a model which takes a scaler and tensor as input and multiplies them and return a tensor. How to do that?
关闭于 2023-06-14 2 条评论