小明想用lambda函数计算两个数的平方和，正确的写法是？（ ）
lambda a,b: a**2 + b**2
lambda a,b: return a**2 + b**2
lambda a,b -> a**2 + b**2
lambda (a,b): a**2 + b**2