tabensemb.model.AbstractNN.to_empty

method

AbstractNN.to_empty(*, device: str | device) → 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