天天看點

pytorch中mm()函數的用法

x = x.mm(self.w) #x與w相乘
           

注:x必須是tensor,才可以應用該方法。

參考連結:

https://blog.csdn.net/genous110/article/details/87801605

繼續閱讀