這是一個(gè)計(jì)算兩個(gè)tensor之間乘積的函數(shù)，該函數(shù)要求兩個(gè)tensor必須都是三維的，且對(duì)兩個(gè)tensor維度有限制，要求啊a，b兩個(gè)tensor有如下格式：
a:(z,x,y)
b:(z,y,c)
則result = torch.bmm(a,b),維度為：（z,x,c）

這里大家應(yīng)該對(duì)該函數(shù)有了一個(gè)更加具體的認(rèn)識(shí)了，還有其他的tensor乘法操作，可以參考我之前的博客。

總結(jié)

以上是生活随笔為你收集整理的pytorch 之 torch.bmm()函数的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。