如何在Tensorflow Java中更改变量的形状？

Question

我有一个矩阵（Tensorflow变量），其形状是动态确定的。所以我需要在创建图形时重新分配矩阵。在python中，我们通过将validate_shape设置为False来options of resetting the graph or reassigning the variable。

代码段：

在张量流图中嵌入矩阵：

embedding_matrix = tf.get_variable("EMB_MATRIX",
                                        shape=[vocab_size, 300],
                                        dtype=tf.float32,
                                        initializer=tf.random_uniform_initializer(-0.1, 0.1, dtype=tf.float32),
                                        trainable=False)

使用新形状重新分配嵌入矩阵：

word_emb_matrix = np.insert(word_emb_matrix, vocab_size, np.arange(300), axis=0) # Inserting one row into embedding matrix
session.run(tf.assign(mtrain.embedding_matrix, word_emb_matrix, name="EMB_MATRIX", validate_shape=False)) #Assign embedding matrix with new shape

现在，当我们在Java中使用保存的模型/图形时，如何实现相同的功能？