Shape check input

Only Cuda and Tensor had C code with check on the input c_extract.
This line wasn't updated.