add custom shape typings
Merge pull request #3306 from qinfchen/sigma_customShape

add custom shape typings for sigma