Zjhao666的博客

原创 一个关于pytorch的tensor点乘的小问题

事情的缘由是，坐旁边的学姐有一段代码将两个二维数组相乘。特别的是，既不是点乘，也不是矩阵乘法，而是将各自每一行分别相乘再拼接得到一个三维数组，具体代码大致如下import torcha = torch.Tensor(range(6)).reshape(2, 3)b = torch.Tensor(range(1, 7)).reshape(2, 3)batch = len(a)lengt...