Underrstanding matrix multiplication for 3D tensors #481
Unanswered
sumitfromculinda
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hi As suggested that for matrix multiplication the input dimension should match.
But when I try the below, I am getting an error:
new_tensor=torch.matmul(torch.rand(224,224,3),torch.rand(3,224,224))
Below is the error:
RuntimeError: The size of tensor a (224) must match the size of tensor b (3) at non-singleton dimension 0
Please suggest why we are getting this error.
Beta Was this translation helpful? Give feedback.
All reactions