search

apache / singa

a distributed deep learning platform
Apache License 2.0
3.37k stars 1.25k forks source link

Extend the matrix multiplication operator to more dimensions #1211

Closed gzrp closed 2 months ago

gzrp commented 2 months ago

Extend the matrix multiplication operator to more dimensions for the transformer example https://github.com/apache/singa/tree/master/examples/trans