cascade.nn.MultiLinear

class cascade.nn.MultiLinear(in_features, out_features, multi_dims)[source]

Bases: Module

Linear layer with support for multi-dims

Parameters:

  • in_features (int) – Input dimensionality

  • out_features (int) – Output dimensionality

  • multi_dims (tuple[int, ...]) – Multiplex dims at the front of input samples

Methods

forward