torch.matmul()

记录一下吧
比如特征x的大小是[5,48,2048] , x2的特征大小是[5,2048,48]

x3 = torch.matmul(x1,x2)

出来结果就是 x3的大小是 [5, 48,48]

你可能感兴趣的:(代码问题,深度学习,python,人工智能)