kevkid

import torch
import numpy as np
seed = 2
torch.manual_seed(seed)
np.random.seed(seed)

x_img = torch.tensor([8,1,3,0,2])
x_txt = torch.tensor([ 0,  7, 10,  1,  1])
out_dim = 3