Встраивание списка реальных значений с помощью TensorFlow

Question

Когда используешь tf.estimator.DNNClassifierи у меня есть feature_a лайки (первое измерение - размер партии, второе - длина одной функции).

import numpy as np

feature_a = np.array([
    [1, 1, 5, 0, 10],
    [0, 1, 0, 0, 1],
    [1, 0, 0, 1, 1],
    [0, 0, 12, 0, 11],
    [1, 0, 1, 0, 1]
])

Вопрос в том, как использовать tf.feature_column.embedding_column встроить это feature_a (Второе измерение feature_a может-достигнет миллионного числа)? Я старался

import tensorflow as tf

feature_a = tf.feature_column.numeric_column(
    key='feature_a', dtype=tf.int32, shape=(5, 5)
)
tf.feature_column.embedding_column(feature_a, dimension=10)

Проблема в том что embedding_column просто прими _CategoricalColumn