# u [32,30,200]
# u_logits [400,32,30]
q_j_400 = []
for j in range(400):
q_j_400.append(tf.squeeze(tf.matmul(tf.transpose(u,[0,2,1]),tf.expand_dims(tf.nn.softmax(u_logits[j]),-1)),[<
tensorflow 用矩阵运算替换for循环 用tf.tile而不写for
最新推荐文章于 2024-04-03 18:34:38 发布
本文介绍了如何利用TensorFlow的tf.tile函数替代for循环,实现更高效的矩阵运算,从而提高代码执行速度。
摘要由CSDN通过智能技术生成