假设我有
R_cap_i_ff = tf.get_variable('R_cap_i_ff', [batch_size, n_components])
R_cap_j_ff = tf.get_variable('R_cap_j_ff', [batch_size, n_components])
这些是2个变量,在Python和TF图中具有相同的名称。
然后我
x = tf.einsum('ij,ij->i', R_cap_i_ff, tf.transpose(R_cap_j_ff))
我可以向x
提供tensorflow名称以及如何执行此操作吗?