MNistNet.to_empty

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

previous

MNistNet.to

next

MNistNet.train