亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

使用動態形狀廣播 tf.matmul

使用動態形狀廣播 tf.matmul

陪伴而非守候 2021-09-25 16:53:58
我想tf.matmul在 2 級和 3 級的兩個張量之間廣播一個操作,其中一個包含“未知”形狀的維度(基本上是特定維度中的“無”值)。問題是,與動態維度tf.reshape和tf.broadcast_to似乎并不工作。x = tf.placeholder(shape=[None, 5, 10], dtype=tf.float32)w = tf.ones([10, 20])y = x @ wwith tf.Session() as sess:  r1 = sess.run(y, feed_dict={x: np.ones([3, 5, 10])})  r2 = sess.run(y, feed_dict={x: np.ones([7, 5, 10])})以上面的代碼為例。在這種情況下,我將分別喂入 3 個和 7 個元素的兩個不同批次。我希望r1并r2成為w這些批次中 3 或 7 個元素中的每一個的矩陣乘法的結果。因此,r1和的結果形狀r2分別為 (3, 5, 20) 和 (7, 5, 20),但我得到:ValueError: Shape must be rank 2 but is rank 3 for 'matmul' (op: 'MatMul') with input shape: [?,5,10], [10,20]。
查看完整描述

2 回答

  • 2 回答
  • 0 關注
  • 260 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號