to_empty
- FullyConnected.to_empty(*, device: Union[str, torch.device]) torch.nn.modules.module.T
Moves the parameters and buffers to the specified device without copying storage.
- Parameters
device (
torch.device
) – The desired device of the parameters and buffers in this module.- Returns
self
- Return type
Module