我正在进行卷积扩张实验，在该过程中，我尝试使用PyTorch将数据从一个2D张量复制到另一个2D张量。我复制从张值A到张量B，使得每一个元素A被复制到B被包围n零。
我已经尝试过使用嵌套for循环，这是一种非常幼稚的方法。当我使用大量灰度图像作为输入时，性能显然很差。
for i in range(A.shape[0]):
for j in range(A.shape[1]):
B[n+i][n+j] = A[i][j]
是否有不需要循环使用的更快的东西？