Генерация матрицы в 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)