Facing Tensor Size issue in Tranformer tool

Happy to help. :blush:. Please mask as solved.