python - 创建随机可训练索引，以从 tensorflow 中的一个 2_D 张量构建两个 2_D 张量

Question

我正在尝试从一个 shape=(30 的张量 AAA 创建两个形状为 shape=(30, 4200) 的张量，4201)并且类型= float32。

运行经过多次不同更改的代码后，此错误将再次出现。

如何选择正确的张量形状？

Answer 1

这就是您在案例中使用 tf.gather_nd 的方式:

# If V1_emb and V2_emb have different shapes you need to take the shape of each one
s = tf.shape(V1, out_type=V1_emb.dtype)
batch_idx = tf.tile(tf.expand_dims(tf.range(s[0]), 1), [1, s[1]])
V1 = tf.gather_nd(params=AAA, indices=tf.stack([batch_idx, V1_emb], axis=-1))
V2 = tf.gather_nd(params=AAA, indices=tf.stack([batch_idx, V2_emb], axis=-1))