/// <summary>
/// Moves the parameters and buffers.
/// </summary>
/// <param name="deviceType">The device type, e.g. 'CPU' or 'CUDA'.</param>
/// <param name="deviceIndex">The optional device index.</param>
/// <returns></returns>
public Module to(DeviceType deviceType, int deviceIndex = -1)
{
Torch.InitializeDeviceType(deviceType);
THSNN_Module_to_device(handle, (int)deviceType, deviceIndex);
Torch.CheckForErrors();
return(this);
}
