Hi,
Given m*n input matrix (tensor),
A B
C D

I want to repeat it diagonally, into (m*p)*(n+p-1) matrix
A B 0 0
0 A B 0
0 0 A B
C D 0 0
0 C D 0
0 0 C D

Where (m,n,p) are provided during init().
How to do it in Pytorch with autograd support? Thank you!