k_cast function

Casts a tensor to a different dtype and returns it.