怎么在指定轴上增加维度?


#1

想请教一下,怎么在指定轴上增加维度啊,有squeeze却没有unsqueeze这个函数啊


#2

expand_dims


#3

那有办法实现像torch.matmul一样的多维矩阵乘法吗


#4

我不知道matmul怎么用的。。。
不过mxnet这边有dot和broadcast_mul,一个是矩阵乘法,一个是对应位置乘法(支持broadcast),你可以去看一下文档这两个里面是不是有你想要的。


#5

嗯,看了一下这两个好像还是不行,就是想实现这样的乘法
例如:如果tensor1是J×1×N×M张量和tensor2是K×M×P张量,tensor1*tensor2将是一个J×K×N×P张量。


#6

mxnet里目前没有和tf.matmul一样的操作符,mxnet上类似的操作符是batch_dot

旧版本的batch_dot仅仅支持维度等于3的tensor,现在官方文档显示最新版本的mxnet支持对维度大于3的tensor使用batch_dot


#7

好的,谢谢