Dependent type-systems like Agda can do that. Unfortunately they are also a pain to write. Fun-fact, you can use church-encoding to transform from your generic

Dependent type-systems like Agda can do that. Unfortunately they are also a pain to write. Fun-fact, you can use church-encoding to transform from your generic int to a "shape" based encoding for the slowest operations ever in quite a few languages (Python included).