Method inputs

ci_test/common_python/test_util.py:178–188

Marks the given tensor as an input of the tested LBANN model, and returns a matching LBANN Input node (or a Slice/Reshape thereof). :param tensor: The input NumPy array to use. :return: An LBANN layer object that will serve as the input.

(self, tensor: Any)

    extra_callbacks: List[lbann.Callback] = field(default_factory=list)

    def inputs(self, tensor: Any) -> lbann.Layer:
        """
        Marks the given tensor as an input of the tested LBANN model, and
        returns a matching LBANN Input node (or a Slice/Reshape thereof).

        :param tensor: The input NumPy array to use.
        :return: An LBANN layer object that will serve as the input.
        """
        self.input_tensor = tensor
        inp = lbann.Input(data_field='samples')
        return slice_to_tensors(inp, tensor)

    def inputs_like(self, *tensors) -> List[lbann.Layer]:
        """