I’m looking for an efficient way to take the outer product of each row of a matrix, e.g.: julia> A = [1 2;4 8] 2×2 Matrix{Int64}: 1 2 4 8 julia> cat([r * r' for r in eachrow(A)]..., dims=3) 2×2×2 Array{Int64, 3}: [:, :, 1] = 1 2 2 4 [:, :, 2] = 16 32 32 64 julia> The number of rows…

Outer product of each row of a matrix

bertschi October 25, 2024, 10:10pm 2

stack(r * r' for r in eachrow(A))

2 Likes

Topic		Replies	Views
Outer broadcasting? General Usage	11	2035	April 5, 2019
Fast computation of row-wise Kronecker product (Khatri-Rao product) Performance question , performance	5	1335	April 8, 2021
How to broadcast or batch multiply a "batch" of matrices with another matrix on the GPU? GPU	1	834	August 29, 2021
Would matrix multiplication benefit if Julia allowed for row-major arrays as well? Internals & Design question , proposal , column-major	1	922	September 20, 2018
Yet another broadcast question General Usage first-steps	10	778	November 15, 2018