explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-15 06:09:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix char_embed for gpu

Browse Source
This commit is contained in:
Matthw Honnibal 2019-10-07 16:49:32 +02:00
parent f4040a98f0
commit a6a2ff217f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/_ml.py
View File

@ -932,7 +932,7 @@ class CharacterEmbed(Model):
# for the tip. # for the tip.
nCv = self.ops.xp.arange(self.nC) nCv = self.ops.xp.arange(self.nC)
for doc in docs: for doc in docs:
doc_ids = doc.to_utf8_array(nr_char=self.nC) doc_ids = self.ops.asarray(doc.to_utf8_array(nr_char=self.nC))
doc_vectors = self.ops.allocate((len(doc), self.nC, self.nM)) doc_vectors = self.ops.allocate((len(doc), self.nC, self.nM))
# Let's say I have a 2d array of indices, and a 3d table of data. What numpy # Let's say I have a 2d array of indices, and a 3d table of data. What numpy
# incantation do I chant to get # incantation do I chant to get