Class BasicInferenceObservable

    • Field Detail

    • Constructor Detail

      • BasicInferenceObservable

        public BasicInferenceObservable​(org.nd4j.linalg.api.ndarray.INDArray... inputs)
      • BasicInferenceObservable

        public BasicInferenceObservable​(org.nd4j.linalg.api.ndarray.INDArray[] inputs,
                                        org.nd4j.linalg.api.ndarray.INDArray[] inputMasks)
    • Method Detail

      • addInput

        public void addInput​(@NonNull
                             @NonNull org.nd4j.linalg.api.ndarray.INDArray... input)
        Specified by:
        addInput in interface InferenceObservable
      • addInput

        public void addInput​(@NonNull
                             @NonNull org.nd4j.linalg.api.ndarray.INDArray[] input,
                             org.nd4j.linalg.api.ndarray.INDArray[] inputMasks)
        Specified by:
        addInput in interface InferenceObservable
      • getInputBatches

        public List<org.nd4j.common.primitives.Pair<org.nd4j.linalg.api.ndarray.INDArray[],​org.nd4j.linalg.api.ndarray.INDArray[]>> getInputBatches()
        Description copied from interface: InferenceObservable
        Get input batches - and their associated input mask arrays, if any
        Note that usually the returned list will be of size 1 - however, in the batched case, not all inputs can actually be batched (variable size inputs to fully convolutional net, for example). In these "can't batch" cases, multiple input batches will be returned, to be processed
        Specified by:
        getInputBatches in interface InferenceObservable
        Returns:
        List of pairs of input arrays and input mask arrays. Input mask arrays may be null.
      • checkOutputException

        protected void checkOutputException()