Я пытаюсь сделать Op, который не ведет себя так, как ожидалось.
graph = tf.Graph()
with graph.as_default():
train_dataset = tf.placeholder(tf.int32, shape=[128, 2])
embeddings = tf.Variable(
tf.random_uniform([50000, 64], -1.0, 1.0))
embed = tf.nn.embedding_lookup(embeddings, train_dataset)
embed = tf.reduce_sum(embed, reduction_indices=0)
Поэтому мне нужно знать размеры тензора embed
. Я знаю, что это можно сделать во время работы, но это слишком много для такой простой операции. Какой более простой способ сделать это?