Interface TensorMerger<T extends Tensor<T>>


public interface TensorMerger<T extends Tensor<T>>
Merged multiple input tensors into a single output which can be processed by a Function.
  • Method Summary

    Modifier and Type Method Description
    void combine​(List<T> inputs, T output)  
    int[] getOutputShape()  
    void initialize​(List<int[]> inputShapes)  
  • Method Details

    • initialize

      void initialize​(List<int[]> inputShapes)
    • combine

      void combine​(List<T> inputs, T output)
    • getOutputShape

      int[] getOutputShape()