python - 如何在 tensorflow keras中重复矩阵？

Question

我正在尝试在tensorflow、keras中重写以下python代码。但是，我在tensorflow keras中找不到等效的np.matlib.repmat。谁能帮我解决这个问题吗？

index = np.arange(0,256);
index_transpose = index.reshape(256,1)

I = numpy.matlib.repmat(index_transpose,1,256).reshape(256,256);
J = numpy.matlib.repmat(index,256,1);

I 和 J 的形状应为:I 形:(256, 256)，J 形:(256, 256)

Answer 1

你可以这样做:

index = K.arange(256) #[0,1,2...,255]

I = K.stack([index]*256, axis=-1) 
J = K.stack([index]*256, axis= 0)

地点:

I = [0,0,0....]
    [1,1,1,....]
    ...
    [255,.....]

J = [0,1,2,3...,255]
    [0,1,2,3...,255]
    .....