error: ‘module’ object has no attribute ‘_rebuild_tensor_v2’

import torch._utils

try:

     torch._utils._rebuild_tensor_v2

except AttributeError:

     def _rebuild_tensor_v2(storage, storage_offset, size, stride, requires_grad, backward_hooks):

                 tensor = torch._utils._rebuild_tensor(storage, storage_offset, size, stride)

                 tensor.requires_grad = requires_grad

                 tensor._backward_hooks = backward_hooks

                 return tensor

   torch._utils._rebuild_tensor_v2 = _rebuild_tensor_v2