示例#1
0
文件: TestNumpy.java 项目: mrj0/jep
  /**
   * Called from python to verify that a Java method's return type of NDArray can be auto-converted
   * to a numpy ndarray.
   *
   * @param array
   * @return a copy of the data + 5
   */
  public NDArray<int[]> testArgAndReturn(NDArray<int[]> array) {
    int[] data = array.getData();
    int[] newData = new int[data.length];
    for (int i = 0; i < data.length; i++) {
      newData[i] = data[i] + 5;
    }

    return new NDArray<int[]>(newData, array.getDimensions());
  }