dqs.torch.layer.SigSoftmax ================================================ SigSoftmax layer for PyTorch. .. code-block:: python class SigSoftmax() Tensor Shape --------------------------------- ==================== ============================================================== I/O Shape ==================== ============================================================== Input (N, M) Output (N, M) ==================== ============================================================== N corresponds the batch size. Example --------------------------------- The following code transforms input tensor x into output tensor y of the same size. .. code-block:: python sizes = torch.tensor([4,4], dtype=torch.long) l = SigSoftmax() x = torch.randn(128, 10) y = l(x)