ResBlock1D.to_empty
- ResBlock1D.to_empty(*, device: int | str | device | None, recurse: bool = True) T[source]
Move the parameters and buffers to the specified device without copying storage.
- Args:
- device (
torch.device): The desired device of the parameters and buffers in this module.
- recurse (bool): Whether parameters and buffers of submodules should
be recursively moved to the specified device.
- device (
- Returns:
Module: self