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

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

總結(jié)

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

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