使用最大池化层降低张量的维数

Question

我的问题很简单：

如何使用max-pooling layer将列表或张量的维度减少到列表中的512个元素：

我正在尝试以下代码：

    input_ids = tokenizer.encode(question, text)
    print(input_ids) # input_ids is a list of 700 elements
    m = nn.AdaptiveMaxPool1d(512)
    input_ids = m(torch.tensor([[input_ids]])) # convert the list to tensor and apply max-pooling layer

但我得到以下错误：

RuntimeError: "adaptive_max_pool2d_cpu" not implemented for 'Long'

因此，请帮助找出错误所在

Answer 1

问题出在你的input_ids上。您正在向AdaptiveMaxPool1d传递一个long类型的张量，只需将其转换为float即可。

    input_ids = tokenizer.encode(question, text)
    print(input_ids) # input_ids is a list of 700 elements
    m = nn.AdaptiveMaxPool1d(512)
    input_ids = m(torch.tensor([[input_ids]]).float()) #