The output should be Tensor2 instead of Tensor2s.
The output should be Tensor2 instead of Tensor2s.