qualia_codegen_core.graph.layers.TPermuteLayer module

class qualia_codegen_core.graph.layers.TPermuteLayer.TPermuteLayer(input_shape: qualia_codegen_core.typing.Shapes, output_shape: qualia_codegen_core.typing.Shapes, output_dtype: qualia_codegen_core.typing.DTypes, name: str, dims: tuple[int, ...])[source]

Bases: TBaseLayer

dims: tuple[int, ...]

input_shape: Shapes

output_shape: Shapes

output_dtype: DTypes

name: str