MnistModel.to_empty#

MnistModel.to_empty(*, device: Union[str, device]) T#

Moves 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.

Returns:

Module: self