tensorflow多維張量計(jì)算實(shí)例
兩個(gè)三維矩陣的乘法怎樣計(jì)算呢?我通過(guò)實(shí)驗(yàn)發(fā)現(xiàn),tensorflow把前面的維度當(dāng)成是batch,對(duì)最后兩維進(jìn)行普通的矩陣乘法。也就是說(shuō),最后兩維之前的維度,都需要相同。
首先計(jì)算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))
運(yùn)行結(jié)果:

計(jì)算結(jié)果表明,兩個(gè)三維矩陣相乘,對(duì)應(yīng)位置的最后兩個(gè)維度的矩陣乘法。
再驗(yàn)證高維的張量乘法:
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))
運(yùn)行結(jié)果:

以上這篇tensorflow多維張量計(jì)算實(shí)例就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
在python中寫(xiě)個(gè)自定義數(shù)據(jù)包協(xié)議的打包和解包測(cè)試
這篇文章主要介紹了在python中寫(xiě)個(gè)自定義數(shù)據(jù)包協(xié)議的打包和解包測(cè)試,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-09-09
Python如何檢驗(yàn)樣本是否服從正態(tài)分布
這篇文章主要介紹了Python如何檢驗(yàn)樣本是否服從正態(tài)分布問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-02-02
python grpc實(shí)現(xiàn)異步調(diào)用(不用grpc異步接口)
grpc同步調(diào)用更簡(jiǎn)單,但是在處理復(fù)雜任務(wù)時(shí),會(huì)導(dǎo)致請(qǐng)求阻塞,影響吞吐,本文主要介紹了python grpc實(shí)現(xiàn)異步調(diào)用,不用grpc異步接口,具有一定的參考價(jià)值,感興趣的可以了解一下2024-04-04
Python實(shí)現(xiàn)推送百度鏈接的示例代碼
有時(shí)為了提高搜索效率,也讓搜索引擎更容易發(fā)現(xiàn)自己的文章,我們需要將文章鏈接推送到百度站長(zhǎng)平臺(tái),起到快速收錄的目的。本文將主要介紹如何通過(guò)Python實(shí)現(xiàn)這一功能,需要的可以參考一下2021-12-12
Python視頻剪輯合并操作的實(shí)現(xiàn)示例
很多人在創(chuàng)作視頻時(shí)都需要進(jìn)行剪輯,本文主要介紹了Python視頻剪輯合并操作的實(shí)現(xiàn)示例,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2025-01-01

