Skip to main content

projection_parameter_group

Function projection_parameter_group 

Source
pub fn projection_parameter_group<T, M>(
    logical_name: impl Into<String>,
    role: ParameterRole,
    module: &M,
    placement: ProjectionSharding,
) -> Result<ParameterGroupSpec, ParallelPlanError>
where T: Tensor, M: Parameterized<T>,
Expand description

Describes one affine projection and all encoding companions.