这是一个计算两个tensor之间乘积的函数,该函数要求两个tensor必须都是三维的,且对两个tensor维度有限制,要求啊a,b两个tensor有如下格式:
a:(z,x,y)
b:(z,y,c)
则result = torch.bmm(a,b),维度为:(z,x,c)
这里大家应该对该函数有了一个更加具体的认识了,还有其他的tensor乘法操作,可以参考我之前的博客。
这是一个计算两个tensor之间乘积的函数,该函数要求两个tensor必须都是三维的,且对两个tensor维度有限制,要求啊a,b两个tensor有如下格式:
a:(z,x,y)
b:(z,y,c)
则result = torch.bmm(a,b),维度为:(z,x,c)
这里大家应该对该函数有了一个更加具体的认识了,还有其他的tensor乘法操作,可以参考我之前的博客。