Method make_reference

ci_test/common_python/test_util.py:209–238

Marks the given tensor as a reference output of the tested LBANN model, and returns a matching LBANN node. :param ref: The reference NumPy array to use. :return: An LBANN layer object that will serve as the reference.

(self, ref: Any)

        return slice_to_tensors(x, *tensors)

    def make_reference(self, ref: Any) -> lbann.Input:
        """
        Marks the given tensor as a reference output of the tested LBANN model,
        and returns a matching LBANN node.

        :param ref: The reference NumPy array to use.
        :return: An LBANN layer object that will serve as the reference.
        """
        # The reference is the second part of the input "samples"
        refnode = lbann.Input(data_field='samples')
        if self.input_tensor is None:
            raise ValueError('Please call ``inputs`` or ``inputs_like`` prior '
                             'to calling ``make_reference`` for correctness.')
        mbsize = self.input_tensor.shape[0]

        # Obtain reference
        refnode = lbann.Reshape(lbann.Identity(
            lbann.Slice(
                refnode,
                slice_points=[
                    numel(self.input_tensor) // mbsize,
                    (numel(self.input_tensor) + numel(ref)) // mbsize
                ],
            )),
                                dims=ref.shape[1:])

        # Store reference
        self.reference = refnode
        self.reference_tensor = ref
        return self.reference

    def set_check_gradients_tensor(self, tensor: lbann.Layer):
        """