Method to

python/oneflow/nn/modules/module.py:1446–1451

(
        self: T,
        device: Optional[Union[int, str, flow.device]] = ...,
        dtype: Optional[flow.dtype] = ...,
    )

    @overload
    def to(
        self: T,
        device: Optional[Union[int, str, flow.device]] = ...,
        dtype: Optional[flow.dtype] = ...,
    ) -> T:
        ...

    @overload
    def to(self: T, dtype: flow.dtype) -> T: