mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
Summary: Some operators, e.g., SoftmaxWithLoss, returns scalar-typed tensor. This would allow us to use those ops without having to write layer manually. Reviewed By: xianjiec, kennyhorror Differential Revision: D4703982 fbshipit-source-id: f33969971c57fc037c9b44adb37af1caba4084b6 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| batch_lr_loss.py | ||
| concat.py | ||
| dot_product.py | ||
| expand_dims.py | ||
| fc.py | ||
| functional.py | ||
| layers.py | ||
| simple_operator_layers.py | ||
| sparse_lookup.py | ||
| sparse_to_dense.py | ||
| split.py | ||
| tags.py | ||