A shape of [200, 200, 2] for both tensors is incompatible for a matrix multiplication.
The docs 90 explain the shape requirements.
Assuming the first dimension is the batch size, you would have to make sure the shapes are defined as:
C = A @ B
[k, m, p] = [k, m, n] @ [k, n, p]