pytorch：如何在两个维度上将参数矩阵重复成更大的矩阵？

Question

转换2D参数张量的最简单语法是什么

A   B
C   D

进入

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

请注意，它们是参数张量，因此我需要autograd将传播梯度从后者扩展到前者。 谢谢！

Answer 1

我在最新的pytorch（1.1）中找到了一个类似于numpy.repeat（）的函数，但需要调用两次：

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

Answer 2

使用einops（相同的代码适用于numpy和pytorch）：

z = einops.repeat(x, 'i j -> (i 2) (j 2)')