UninitializedTensorMixin - requires_grad allowed?

tjoseph · January 4, 2024, 12:09pm

Hi

I am wondering why requires_grad_ is not in _allowed_methods? Is there a reason or has it just been forgotten?

pytorch/pytorch/blob/91bbcf8c718ff245ec85e33874dc4c1a1a28c4f3/torch/nn/parameter.py#L86


      
          
                  return (
                      torch._utils._rebuild_parameter_with_state,
                      (self.data, self.requires_grad, hooks, state)
                  )
          
              __torch_function__ = _disabled_torch_function_impl
          
          
          class UninitializedTensorMixin:
              _allowed_methods = [
                  torch.Tensor.__hash__,
                  torch.Tensor.size,
                  torch.Tensor.copy_,
                  torch.Tensor.is_floating_point,
                  torch.Tensor.half,
                  torch.Tensor.float,
                  torch.Tensor.double,
                  torch.Tensor.char,
                  torch.Tensor.short,
                  torch.Tensor.int,