Thanks for your reply.
The error is as follow.
Traceback (most recent call last):
File “ioucompute.py”, line 199, in mapping_mat = torch.matmul(frame, scaling_mat)
RuntimeError: Expected tensor to have size 256 at dimension 1, but got size 3 for argument #2 ‘batch2’ (while checking arguments for bmm)