pytorchtorch matmul between 2D a

pytorch - torch matmul between 2D and 3D tensor - Stack Overflow
Hi I have two tensors:a = torch.randn(125, 128)# Shape: (125, 128)b = torch.randn(128, 8, 64)#
pytorchtorch matmul between 2D and 3D tensorStack Overflow
admin2天前
30