我在tensorflow中有两个张量,第一个张量是3D的,第二个是2D的。我想要像这样将它们相乘:
x = tf.placeholder(tf.float32, shape=[sequence_length, batch_size, hidden_num])
w = tf.get_variable("w", [hidden_num, 50])
b = tf.get_variable("b", [50])
output_list = []
for step_index in range(sequence_length):
output = tf.matmul(x[step_index, :, :], w) + b
output_list.append(output)
output = tf.pack(outputs_list)
我使用循环进行乘法运算,但我觉得这样太慢了。有什么最简单/清晰的方法可以使这个过程更快?
tf.matmul
而不是tf.batch_matmul
。 - holdenlee