Explicitly define the RNG state tensor CPU tensor to avoid failure in CUDA device context

Explicitly define the RNG state tensor CPU tensor to avoid failure in CUDA device context #2514