decrypt_to_pytorch_tensor

Engine.decrypt_to_pytorch_tensor(ciphertext, secret_key)

암호문을 PyTorch tensor의 형태로 복호화합니다. 엔진의 모드가 GPU인 경우, 반환되는 tensor도 GPU에 할당 됩니다.

• 인풋:
  • Ciphertext
  • SecretKey
• 아웃풋:
  • Plaintext

예시

CPU

import torch

from desilofhe import Engine

engine = Engine()
secret_key = engine.create_secret_key()

tensor = torch.tensor([1, 2, 3])
encrypted = engine.encrypt_pytorch_tensor(tensor, secret_key)
decrypted = engine.decrypt_to_pytorch_tensor(encrypted, secret_key)

GPU

import torch

from desilofhe import Engine

engine = Engine(mode="gpu")
secret_key = engine.create_secret_key()

tensor = torch.tensor([1, 2, 3], device="cuda")
encrypted = engine.encrypt_pytorch_tensor(tensor, secret_key)
decrypted = engine.decrypt_to_pytorch_tensor(encrypted, secret_key)