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[]
     
    void
    initialize(List<int[]> inputShapes)
     
  • Method Details

    • initialize

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

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

      int[] getOutputShape()