I think Futhark approaches this with uniqueness types. A bit strange (maybe that's just me?) but it works well with arrays.

Also it compiles to Python nicely.