deepdow.layers.collapse module¶

Collection of layers that decrease the number of dimensions.

class AttentionCollapse(n_channels)[source]¶

Bases: torch.nn.modules.module.Module

Collapsing over the channels with attention.

affine¶

Fully connected layer performing linear mapping.

context_vector¶

Fully connected layer encoding direction importance.

forward(x)[source]¶

Perform forward pass.

Parameters: x (torch.Tensor) – Tensor of shape (n_samples, n_channels, lookback, n_assets).
Returns: Tensor of shape (n_samples, n_channels, n_assets).
Return type: torch.Tensor

class AverageCollapse(collapse_dim=2)[source]¶

Bases: torch.nn.modules.module.Module

Global average collapsing over a specified dimension.

forward(x)[source]¶

Perform forward pass.

Parameters: x (torch.Tensor) – N-dimensional tensor of shape (d_0, d_1, …, d_{N-1}).
Returns: {N-1}-dimensional tensor of shape (d_0, …, d_{collapse_dim - 1}, d_{collapse_dim + 1}, …, d_{N-1}). Average over the removeed dimension.
Return type: torch.Tensor

class ElementCollapse(collapse_dim=2, element_ix=- 1)[source]¶

Bases: torch.nn.modules.module.Module

Single element over a specified dimension.

forward(x)[source]¶

Perform forward pass.

Parameters: x (torch.Tensor) – N-dimensional tensor of shape (d_0, d_1, …, d_{N-1}).
Returns: {N-1}-dimensional tensor of shape (d_0, …, d_{collapse_dim - 1}, d_{collapse_dim + 1}, …, d_{N-1}). Taking the self.element_ix element of the removed dimension.
Return type: torch.Tensor

class ExponentialCollapse(collapse_dim=2, forgetting_factor=None)[source]¶

Bases: torch.nn.modules.module.Module

Exponential weighted collapsing over a specified dimension.

The unscaled weights are defined recursively with the following rules:

Parameters

collapse_dim (int) – What dimension to remove.
forgetting_factor (float or None) – If float, then fixed constant. If None this will become learnable.

forward(x)[source]¶

Perform forward pass.

Parameters: x (torch.Tensor) – N-dimensional tensor of shape (d_0, d_1, …, d_{N-1}).
Returns: {N-1}-dimensional tensor of shape (d_0, …, d_{collapse_dim - 1}, d_{collapse_dim + 1}, …, d_{N-1}). Exponential Average over the removed dimension.
Return type: torch.Tensor

class MaxCollapse(collapse_dim=2)[source]¶

Bases: torch.nn.modules.module.Module

Global max collapsing over a specified dimension.

forward(x)[source]¶

Perform forward pass.

Parameters: x (torch.Tensor) – N-dimensional tensor of shape (d_0, d_1, …, d_{N-1}).
Returns: {N-1}-dimensional tensor of shape (d_0, …, d_{collapse_dim - 1}, d_{collapse_dim + 1}, …, d_{N-1}). Maximum over the removed dimension.
Return type: torch.Tensor

class SumCollapse(collapse_dim=2)[source]¶

Bases: torch.nn.modules.module.Module

Global sum collapsing over a specified dimension.

forward(x)[source]¶

Perform forward pass.

Parameters: x (torch.Tensor) – N-dimensional tensor of shape (d_0, d_1, …, d_{N-1}).
Returns: {N-1}-dimensional tensor of shape (d_0, …, d_{collapse_dim - 1}, d_{collapse_dim + 1}, …, d_{N-1}). Sum over the removed dimension.
Return type: torch.Tensor