code: @Override
      public void partialUpdate(INDArray arr, INDArray result, long idx, int... dimensions) {
          result.tensorAlongDimension((int) idx, dimensions).addi(arr);
      }
docs: Updates result
  based on arr along a particular
  {@link INDArray#tensorAlongDimension(int, int...)}
  
  @param arr        the array to update
  @param result     the result ndarray to update
  @param idx        the index to update
  @param dimensions the dimensions to update