Flex¶
-
class
caldera.blocks.Flex(module_type)[source]¶ Flex.
- Parameters
module_type (
Type[~M]) –
Initialize a module as a FlexBlock with flexible dimensions.
Usage:
Flex(torch.nn.Linear)(Flex.d(), 25)
- Parameters
module_type (
Type[~M]) – module type (e.g. torch.nn.Linear
-
__init__(module_type)[source]¶ Initialize a module as a FlexBlock with flexible dimensions.
Usage:
Flex(torch.nn.Linear)(Flex.d(), 25)
- Parameters
module_type (
Type[~M]) – module type (e.g. torch.nn.Linear
Methods
__init__(module_type)Initialize a module as a FlexBlock with flexible dimensions.
has_flex_blocks(module)- param module
has_unresolved_flex_blocks(module)- param module