使用嵌入层作为编码器的输入

Question

我想使用嵌入层作为编码器的输入，但是我得到了如下错误。输入y是一个形状为1*84的时间序列数据。你能帮我一下吗?

import numpy 
import torch.nn as nn
r_input    = torch.nn.Embedding(84, 10)  
activation  = nn.functional.relu
mu_r      = nn.Linear(10, 6)
log_var_r = nn.Linear(10, 6)
y = np.random.rand(1,84) 
def encode_r(y):
y         = torch.reshape(y, (-1, 1, 84)) # torch.Size([batch_size, 1, 84])
hidden    = torch.flatten(activation(r_input(y)), start_dim = 1)       
z_mu      = mu_r(hidden)
z_log_var = log_var_r(hidden)
return z_mu, z_log_var```
Error: RuntimeError: Expected tensor for argument #1 'indices' to have one of the following scalar types: Long, Int; but got torch.cuda.FloatTensor instead (while checking arguments for embedding)

Answer 1

根据这个线程:https://discuss.pytorch.org/t/expected-tensor-for-argument-1-indices-to-have-scalar-type-long-but-got-cpufloattensor-instead-while-checking-arguments-for-embedding/32441/4，似乎一个可能的解决方案是确保嵌入具有整数值而不是浮点值(通过嵌入，我们指的是查找表而不是实际嵌入向量)。