Combination

class Combination(*args, **kwargs)[source]

Base class for combinations.

Initialize internal Module state, shared by both nn.Module and ScriptModule.

Methods Summary

`forward`(xs)	Combine a sequence of individual representations.
`output_shape`(input_shapes)	Calculate the output shape for the given input shapes.

Methods Documentation

abstract forward(xs: Sequence[Tensor]) → Tensor[source]

Combine a sequence of individual representations.

Parameters:: xs (Sequence[Tensor]) – shape: (*batch_dims, *input_dims_i) the individual representations
Returns:: shape: (*batch_dims, *output_dims) a combined representation
Return type:: Tensor

output_shape(input_shapes: Sequence[tuple[int, ...]]) → tuple[int, ...][source]

Calculate the output shape for the given input shapes.

Note

this method runs a single forward pass if no symbolic computation is available.

Parameters:: input_shapes (Sequence[tuple[int, ...]]) – the input shapes without the batch dimensions
Returns:: the output shape
Return type:: tuple[int, …]