How to convert list of str into Pytorch tensor

Question

How do I convert a list of strings into a string/character tensor in Pytorch?

Related example with numpy:

import numpy as np

mylist = ["this","is","my","list"]

np.array([mylist])

Returns:

array([['this', 'is', 'my', 'list']], dtype='<U4')

However, in pytorch:

torch.tensor(mylist)

Returns:

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-156-36722d81da09> in <module>
----> 1 torch.tensor(mylist)

ValueError: too many dimensions 'str'

A tensor is a multi-dimensional array, so I'm assuming this is possible pytorch.

Note: this post does not answer my question

According to the docs there's no string type tensor.

jodag
– jodag

2019-12-15 08:20:40 +00:00
Commented Dec 15, 2019 at 8:20 — jodag
– jodag, Commented Dec 15, 2019 at 8:20

sujay · Accepted Answer · 2019-12-15 13:28:25Z

6

There is no string tensor so you cannot directly convert to pytorch tensor of strings.

Alternative, you can convert the string to ASCII char values and save that as a Tensor.

answered Dec 15, 2019 at 13:28

sujay

3953 silver badges16 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to convert list of str into Pytorch tensor

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related