Unify the shape notation for all of the pytorch modules (#15741)
authorSasha Rush <srush@seas.harvard.edu>
Thu, 17 Jan 2019 18:04:51 +0000 (10:04 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 17 Jan 2019 18:32:14 +0000 (10:32 -0800)
commitdbe6a7a9ff1a364a8706bf5df58a1ca96d2fd9da
treeb2b72ef08bec01caa0cf6b4dab7564fdd645de1f
parent67f2039f4ce233754910ebc24fbfcc8bc68685ae
Unify the shape notation for all of the pytorch modules (#15741)

Summary:
PR to update the shape notation for all of the torch.nn modules to take a unified form. The goal is to make these definitions machine-readable and those checkable by unifying the style across all of the different modules.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15741

Differential Revision: D13709601

Pulled By: ezyang

fbshipit-source-id: fb89a03903fdf0cd0dcf76f3e469b8582b2f3634
torch/nn/modules/activation.py
torch/nn/modules/adaptive.py
torch/nn/modules/container.py
torch/nn/modules/dropout.py
torch/nn/modules/linear.py
torch/nn/modules/loss.py
torch/nn/modules/normalization.py
torch/nn/modules/pixelshuffle.py
torch/nn/modules/rnn.py
torch/nn/modules/sparse.py