For a problem I’m working on I want to multiply complex matrices together so that I can then for the simplest case is slice out a specific element of this final complex matrix and do the square modulus of that element. However the catch is that PyMC (well specifically Pytensor) models don’t like com…

I might be missing something, but couldn’t you implement it like this? def complex_matmult(A, B): """"Given `A = A + i * Ai` and `B = B + i * Bi` compute the real and imaginary comonents of `AB`""" A, Ai = A B, Bi = B C = A @ B - Ai @ Bi Ci = A @ Bi + Ai @ B return C, Ci U1…

Help formulating a complex matrix multiplication in pymc

Questions v5

ricardoV94 April 19, 2024, 7:52pm 2

CC @aseyboldt

Topic		Replies	Views
Matrix multiplication in pymc3 Questions development	8	6389	April 24, 2018
Matrix-based predictions and complex variables	11	516	February 9, 2024
Create model matrix Questions	7	1518	June 23, 2021
Batch matrix multiplication	2	466	October 13, 2023
Combining parameters into single tensor Questions	19	3538	October 30, 2017

Help formulating a complex matrix multiplication in pymc

Related topics