Robuta

https://caffe2.ai/doxygen-python/html/classtorch_1_1nn_1_1__functions_1_1thnn_1_1sparse_1_1_embedding_bag.html Caffe2 - Python API: torch.nn._functions.thnn.sparse.EmbeddingBag Class Reference python api torch nn