def __init__()

in differentiable_robot_model/rigid_body_params.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, dim1, dim2, init_tensor=None, init_std=0.1):
        super().__init__()
        self._dim1 = dim1
        self._dim2 = dim2
        if init_tensor is None:
            init_tensor = torch.empty(dim1, dim2).normal_(mean=0.0, std=init_std)
        self.param = torch.nn.Parameter(init_tensor)