59d4a8b3bea3b5969f79b45456c8e594b617ce92,gpytorch/lazy/mul_lazy_variable.py,MulLazyVariable,__init__,#MulLazyVariable#,10

Before Change

            - num_samples (int) - the samples number when matmul_mode=stochastic
        """
        super(MulLazyVariable, self).__init__(*lazy_vars, **kwargs)
        if not all([isinstance(lazy_var, LazyVariable) for lazy_var in lazy_vars]):
            raise RuntimeError("All arguments of a MulLazyVariable should be lazy variables")

        self.lazy_vars = lazy_vars

After Change

            - num_samples (int) - the samples number when matmul_mode=stochastic
        """
        lazy_vars = list(lazy_vars)
        for i, lazy_var in enumerate(lazy_vars):
            if not isinstance(lazy_var, LazyVariable):
                if isinstance(lazy_var, Variable):
                    lazy_vars[i] = NonLazyVariable(lazy_var)
                else:
                    raise RuntimeError("All arguments of a SumLazyVariable should be lazy variables or vairables")
        super(MulLazyVariable, self).__init__(*lazy_vars, **kwargs)

        self.lazy_vars = lazy_vars

Instances