diff --git a/prototorch/core/initializers.py b/prototorch/core/initializers.py index 79101f4..d6ae098 100644 --- a/prototorch/core/initializers.py +++ b/prototorch/core/initializers.py @@ -465,7 +465,15 @@ class OnesLinearTransformInitializer(AbstractLinearTransformInitializer): return self.generate_end_hook(weights) -class EyeTransformInitializer(AbstractLinearTransformInitializer): +class RandomLinearTransformInitializer(AbstractLinearTransformInitializer): + """Initialize a matrix with random values.""" + + def generate(self, in_dim: int, out_dim: int): + weights = torch.rand(in_dim, out_dim) + return self.generate_end_hook(weights) + + +class EyeLinearTransformInitializer(AbstractLinearTransformInitializer): """Initialize a matrix with the largest possible identity matrix.""" def generate(self, in_dim: int, out_dim: int): @@ -539,8 +547,9 @@ RRI = RandomReasoningsInitializer ZRI = ZerosReasoningsInitializer # Aliases - Transforms -Eye = EyeTransformInitializer +ELTI = Eye = EyeLinearTransformInitializer OLTI = OnesLinearTransformInitializer +RLTI = RandomLinearTransformInitializer ZLTI = ZerosLinearTransformInitializer PCALTI = PCALinearTransformInitializer LLTI = LiteralLinearTransformInitializer