dagma.locally_connected.LocallyConnected.forward¶

dagma.locally_connected.LocallyConnected.forward(input: torch.Tensor) → torch.Tensor¶

Implements the forward pass of the layer.

Parameters:¶

input : torch.Tensor¶

Shape \((n, d, m1)\)

Returns:¶

Shape \((n, d, m2)\)

Return type:¶

torch.Tensor