Pytorch: как повторить матрицу параметров в большую по обоим измерениям?

Question

Какой самый простой синтаксис для преобразования тензора двумерного параметра

A   B
C   D

в

A  A  B  B
A  A  B  B
C  C  D  D
C  C  D  D

Обратите внимание, что они являются тензорами параметров, поэтому мне нужен автоград для обратного распространения градиента от последнего к прежнему. Спасибо!

Answer 1

Я нашел функцию, похожую на numpy.repeat () в последней версии pytorch (1.1), но ее нужно вызывать дважды:

z = x.repeat_interleave(2,dim=0).repeat_interleave(2,dim=1)