tensorflow多維張量計算實例
更新時間：2020年02月11日 12:26:33 作者：PROMINEM
今天小編就為大家分享一篇tensorflow多維張量計算實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
兩個三維矩陣的乘法怎樣計算呢?我通過實驗發(fā)現(xiàn)，tensorflow把前面的維度當(dāng)成是batch，對最后兩維進行普通的矩陣乘法。也就是說，最后兩維之前的維度，都需要相同。
首先計算shape為(2, 2, 3)乘以shape為(2, 3, 2)的張量。
import tensorflow as tf
import numpy as np
a = tf.constant(np.arange(1, 13, dtype=np.float32), shape=[2, 2, 3])
b = tf.constant(np.arange(1, 13, dtype=np.float32), shape=[2, 3, 2])
c = tf.matmul(a, b)
# c = tf.matmul(a, b)
sess = tf.Session()
print("a*b = ", sess.run(c))
c1 = tf.matmul(a[0, :, :], b[0, :, :])
print("a[1]*b[1] = ", sess.run(c1))
運行結(jié)果：
計算結(jié)果表明，兩個三維矩陣相乘，對應(yīng)位置的最后兩個維度的矩陣乘法。
再驗證高維的張量乘法：
import tensorflow as tf
import numpy as np
a = tf.constant(np.arange(1, 36, dtype=np.float32), shape=[3, 2, 2, 3])
b = tf.constant(np.arange(1, 36, dtype=np.float32), shape=[3, 2, 3, 2])
c = tf.matmul(a, b)
# c = tf.matmul(a, b)
sess = tf.Session()
print("a*b = ", sess.run(c))
c1 = tf.matmul(a[0, 0, :, :], b[0, 0, :, :])
print("a[1]*b[1] = ", sess.run(c1))
運行結(jié)果：
以上這篇tensorflow多維張量計算實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了決策樹的python實現(xiàn)方法,詳細分析了決策樹的優(yōu)缺點及算法思想并以完整實例形式講述了Python實現(xiàn)決策樹的方法,具有一定的借鑒價值,需要的朋友可以參考下2014-11-11
在本篇文章里小編給大家整理的是一篇關(guān)于python中upper的作用的相關(guān)文章，有需要的朋友們可以參考下。2020-07-07
這篇文章主要為大家詳細介紹了如何利用Python語言實現(xiàn)在一行中交換兩個變量功能，文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下2022-08-08
解決pip安裝報錯required?to?install?pyproject.toml-based?projec
這篇文章主要介紹了解決pip安裝報錯required?to?install?pyproject.toml-based?projects問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-05-05
Django與數(shù)據(jù)庫交互的實現(xiàn)
最近在學(xué)習(xí)Django，本文主要介紹了Django與數(shù)據(jù)庫交互的實現(xiàn)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-06-06
一個基于flask的web應(yīng)用誕生 記錄用戶賬戶登錄狀態(tài)（6）
一個基于flask的web應(yīng)用誕生第六篇，這篇文章主要介紹了記錄用戶賬戶登錄狀態(tài)功能開發(fā)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-04-04
Linux 發(fā)郵件磁盤空間監(jiān)控(python)
這篇文章主要介紹了Linux發(fā)郵件磁盤空間監(jiān)控功能，python實現(xiàn)，需要的朋友可以參考下2016-04-04
這篇文章主要為大家介紹了Python中雙下劃線的特殊方法屬性魔法世界探索,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-11-11
sys?模塊是 Python 標準庫中的一個核心模塊,提供了與解釋器進行交互的功能,了解?sys?模塊的方法和變量對于更有效地管理和調(diào)試 Python 程序至關(guān)重要,本文將深入探討?sys?模塊的常用方法和變量,通過詳細的示例代碼,幫助大家更全面地了解并靈活運用這一關(guān)鍵模塊2024-01-01
這篇文章為大家詳細介紹了在圖像上面進行了圖像均值濾波、方框濾波 、高斯濾波、中值濾波、雙邊濾波、2D卷積等具體操作的方法，需要的可以參考一下2023-02-02