Could you check the stack trace and see which function is raising this error?
Once you’ve isolated this function, could you post the shapes of all input tensors, so that we can have a look why this issue is raised?
If you cannot isolate the function, which raises the error, please post the complete stack trace here.