根据行和列在Keras中生成矩阵

Question

我如何在 Keras 中创建一个层来输出给定维度（例如 m、n）的矩阵，其中的单元格具有基于行和列的值？这是论坛：

A[i, 2j] = i / (10**(2*j))
A[i, 2j+1] = i / (10**(2*j))

我试图查看 lamba 函数，但 Keras 似乎只传递了单元格值而不是索引！任何其他选项（不是循环）

Answer 1

您可以执行以下操作：

from keras.layers import Input
import keras.backend as K
import numpy as np


def CustomConstantInput(m, n):
    x = np.arange(m)
    y = 10 ** (2 * (np.arange(n) // 2))
    matrix = x[:, None] / y[None, :]
    print(matrix)
    fixed_input = K.constant(matrix)
    return Input(tensor=fixed_input)


t = CustomConstantInput(3, 4)

根据行和列在Keras中生成矩阵

Generate matrix in Keras based on row and column

lambda

matrix

layer

keras