trw.layers.utils

Module Contents

Functions

div_shape(shape: Union[Sequence[int], int], div: int = 2) → Union[Sequence[int], int]

Divide the shape by a constant

trw.layers.utils.div_shape(shape: Union[Sequence[int], int], div: int = 2) → Union[Sequence[int], int]

Divide the shape by a constant

Parameters

• shape – the shape

• div – a divisor

Returns

a list