Matrix Product: PyTorch Practice Problem (Easy)

13. Matrix Product

Easy

Return the matrix product of A and B.

Input

A =
tensor([[1., 1., 1.],
        [1., 1., 1.],
        [1., 1., 1.],
        [1., 1., 1.],
        [1., 1., 1.]])
B =
tensor([[1., 1.],
        [1., 1.],
        [1., 1.]])

Output

tensor([[3., 3.],
        [3., 3.],
        [3., 3.],
        [3., 3.],
        [3., 3.]])

Sign in to write and run your own code.

Sign in Create free account

Implement solve(...)