source
espnet2.speechlm.utils.data.to_device(data, device=None, dtype=None, non_blocking=False, copy=False)
Change the device of object recursively