tensorflow 多维矩阵相乘 多维tensor相乘

tensorflow 多维矩阵相乘 多维tensor相乘

import tensorflow as tfsess = tf.Session()left = tf.ones(shape=[16,20])right = tf.ones(shape=[20,100])result = tf.einsum('in,nd->id', left, right)print(sess.run(tf.shape(result)))left = tf.ones(shape=[10,16,20])right = tf.ones(shape=[20,100])result = tf.einsum('ibn,nd->ibd', left, right)print(sess.run(tf.shape(result)))left = tf.ones(shape=[10,16,20])right = tf.ones(shape=[20,24,100])result = tf.einsum('ibh,hnd->ibnd', left, right)print(sess.run(tf.shape(result)))

print结果 [ 16 100] [ 10 16 100] [ 10 16 24 100]

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。