想请教一下,怎么在指定轴上增加维度啊,有squeeze却没有unsqueeze这个函数啊
怎么在指定轴上增加维度?
snow6666667
#4
我不知道matmul怎么用的。。。
不过mxnet这边有dot和broadcast_mul,一个是矩阵乘法,一个是对应位置乘法(支持broadcast),你可以去看一下文档这两个里面是不是有你想要的。
wushengdu
#5
嗯,看了一下这两个好像还是不行,就是想实现这样的乘法
例如:如果tensor1是J×1×N×M张量和tensor2是K×M×P张量,tensor1*tensor2将是一个J×K×N×P张量。