Module
type(dst_type)
  Casts all parameters and buffers to "dst_type".

  Note:

    This method modifies the module in-place.

  Parameters:
     **dst_type** (*type** or **string*) -- the desired type

  Returns:
     self

  Return type:
     Module

xpu(device=None)
  Moves all model parameters and buffers to the XPU.

  This also makes associated parameters and buffers different
  objects. So it should be called before constructing optimizer if
  the module will live on XPU while being optimized.

  Note:

    This method modifies the module in-place.

  Parameters:
     **device** (*int**, **optional*) -- if specified, all
     parameters will be copied to that device

  Returns:
     self

  Return type:
     Module

zero_grad(set_to_none=False)
  Sets gradients of all model parameters to zero. See similar
  function under "torch.optim.Optimizer" for more context.

  Parameters:
