Slice¶
Bases: BaseModule
Keep only the first keep_bits elements of the last dimension.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
keep_bits
|
int
|
Number of elements to retain along the last dimension. |
required |