Create the tensor using torch.frombuffer to write shm with tensors in state dict.

Create the tensor using torch.frombuffer to write shm with tensors in state dict. #2642