grain ~master (2021-05-15T05:53:16Z)

Embedding.forward

grain functions binary Embedding
Undocumented in source. Be warned that the author may not have intended to support it.
  1. auto forward(Variable!(T, 2, HostStorage) weight, Variable!(int, 1, HostStorage) ids)
    struct Embedding(T)
  2. auto forward(Variable!(T, 2, DeviceStorage) weight, Variable!(int, 1, DeviceStorage) ids)

Meta

Source

See Implementation

grain functions binary Embedding functionsmixinsvariables

Page generated by adrdox