How can we compute batch matrix-matrix product of matrices (3-D tensors) in MATLAB?

9 次查看（过去 30 天）

BIPIN SAMUEL 2023-12-5

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2056424-how-can-we-compute-batch-matrix-matrix-product-of-matrices-3-d-tensors-in-matlab

评论： BIPIN SAMUEL 2023-12-7

I have two 3-D tensors of size b*n*m and b*m*p. I need the product of these two tensors to get the output size as b*n*p. I have used direct matrix multiplication using * operator. Moreover, I have used 'pagetimes'. But it is not working. Is there any function in MATLAB like 'torch.bmm' in python as shown in link: https://pytorch.org/docs/stable/generated/torch.bmm.html to do the matrix multiplication of two 3-D tensors.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

采纳的回答

Bruno Luong 2023-12-5

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2056424-how-can-we-compute-batch-matrix-matrix-product-of-matrices-3-d-tensors-in-matlab#answer_1366019

在 MATLAB Online 中打开

b = 2;
n = 3;
m = 4;
p = 5;
A = rand(b,n,m);
B = rand(b,  m,p);
AA = permute(A, [2 3 1]);
BB = permute(B, [2 4 1 3]);
AB=pagemtimes(AA, BB);
AB=permute(AB,[3 1 4 2]);
size(AB)
ans = 1×3
     2     3     5
AB
AB = 
AB(:,:,1) =

    1.1069    0.9331    1.2839
    0.4248    1.0635    1.2576


AB(:,:,2) =

    1.3688    1.5129    1.6656
    0.6568    1.2286    1.3592


AB(:,:,3) =

    1.0351    1.2450    1.3362
    0.4743    1.1831    1.3162


AB(:,:,4) =

    1.2485    1.0840    1.4573
    0.7987    1.7168    1.7815


AB(:,:,5) =

    0.5739    0.7361    0.7294
    0.9621    1.9460    2.0458

5 个评论
显示 3更早的评论隐藏 3更早的评论

Bruno Luong 2023-12-6

Sorry I don't know dlarray, and I'm not sure what layer do you refer to.

BIPIN SAMUEL 2023-12-7

I refer to the deep learning layer where input is 3-D tensor in the formatted dlarray and I want to do these operations as a part of forming that layer.

请先登录，再进行评论。