首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何使用pytorch将python列表列表转换为张量

如何使用pytorch将python列表列表转换为张量
EN

Stack Overflow用户
提问于 2021-11-22 14:43:08
回答 1查看 55关注 0票数 0

我得到一个列表,其中包含不同长度的列表。如何在不使用填充的情况下将其转换为pytorch中的张量?有可能吗?

代码语言:javascript
复制
[[3, 5, 10, 11], [1, 5, 10]]
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-11-22 21:42:41

这取决于您想要通过数据结构实现什么。您可以使用torch.sparse,例如:

代码语言:javascript
复制
ll = [[3, 5, 10, 11], [1, 5, 10]]
n = len(ll)
m = max(len(l) for l in ll)

ids = [[], []]
values = []
for i, l in enumerate(ll):
    length = len(l)
    ids[0] += [i] * length  # rows
    ids[1] += list(range(length))  # cols
    values += l

t = torch.sparse_coo_tensor(ids, values, (n, m))

否则,您可以尝试使用文档语料库的嵌入技术,例如bag- of -words (尽管它仍然会生成一些“填充”)、tf-idf等。

内部列表中可能存在重复项的词袋

代码语言:javascript
复制
corpus = [[3, 5, 10, 11], [1, 5, 10]]
n = len(corpus)
m = max(max(inner) for inner in corpus)
t = torch.zeros(n, m)

for i, doc in enumerate(corpus):
    torch.bincount(corpus)

内部列表中具有不同值的词袋

代码语言:javascript
复制
corpus = [[3, 5, 10, 11], [1, 5, 10]]
n = len(corpus)
m = max(max(inner) for inner in corpus)

t = torch.zeros(n, m)
for i, doc in enumerate(corpus):
    t[i, doc] = 1
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70067595

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档