Concat¶
Bases: BaseModule
Concatenate a list of tensors along a non-batch dimension.
Called explicitly (not through :class:torch.nn.Sequential) with a list
of tensors as input.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
dim
|
int
|
Concatenation dimension, given without the batch dimension
( |
required |
input_shapes
|
list[list[int]]
|
Per-input shapes (batch-excluded). All inputs must
match on every dimension except |
required |