When I execute the program mapping_mat = torch.matmul(frame, scaling_mat)，there is a RuntimeError: Expected tensor to have size 256 at dimension 1, but got size 3 for argument #2 ‘batch2’ (while checking arguments for bmm).
frame is torch.Size([1, 3, 256, 256])
scaling_mat is torch.Size([1, 3, 3])