Package ai.djl.tensorflow.engine

Class TfNDArray

java.lang.Object
ai.djl.util.NativeResource<org.tensorflow.internal.c_api.TFE_TensorHandle>
ai.djl.tensorflow.engine.TfNDArray
All Implemented Interfaces:
ai.djl.ndarray.BytesSupplier, ai.djl.ndarray.NDArray, ai.djl.ndarray.NDResource, AutoCloseable
public class TfNDArray extends ai.djl.util.NativeResource<org.tensorflow.internal.c_api.TFE_TensorHandle> implements ai.djl.ndarray.NDArray
TfNDArray is the TensorFlow implementation of NDArray.

  • Field Summary

    Fields inherited from class ai.djl.util.NativeResource

    handle

  • Method Summary

    Modifier and Type
    Method
    Description
    ai.djl.ndarray.NDArray
    abs()
    ai.djl.ndarray.NDArray
    acos()
    ai.djl.ndarray.NDArray
    acosh()
    ai.djl.ndarray.NDArray
    add(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    add(Number n)
    ai.djl.ndarray.NDArray
    addi(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    addi(Number n)
    ai.djl.ndarray.NDArray
    all()
    ai.djl.ndarray.NDArray
    any()
    ai.djl.ndarray.NDArray
    argMax()
    ai.djl.ndarray.NDArray
    argMax(int axis)
    ai.djl.ndarray.NDArray
    argMin()
    ai.djl.ndarray.NDArray
    argMin(int axis)
    ai.djl.ndarray.NDArray
    argSort(int axis, boolean ascending)
    ai.djl.ndarray.NDArray
    asin()
    ai.djl.ndarray.NDArray
    asinh()
    ai.djl.ndarray.NDArray
    atan()
    ai.djl.ndarray.NDArray
    atan2(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    atanh()
    void
    attach(ai.djl.ndarray.NDManager manager)
    ai.djl.ndarray.NDArray
    batchDot(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    batchMatMul(ai.djl.ndarray.NDArray other)
     
    ai.djl.ndarray.NDArray
    booleanMask(ai.djl.ndarray.NDArray index, int axis)
    ai.djl.ndarray.NDArray
    broadcast(ai.djl.ndarray.types.Shape shape)
    ai.djl.ndarray.NDArray
    cbrt()
    ai.djl.ndarray.NDArray
    ceil()
    ai.djl.ndarray.NDArray
    clip(Number min, Number max)
    void
    close()
    ai.djl.ndarray.NDArray
    complex()
    ai.djl.ndarray.NDArray
    conj()
    boolean
    contentEquals(ai.djl.ndarray.NDArray other)
    boolean
    contentEquals(Number number)
    ai.djl.ndarray.NDArray
    cos()
    ai.djl.ndarray.NDArray
    cosh()
    ai.djl.ndarray.NDArray
    cumProd(int axis)
    ai.djl.ndarray.NDArray
    cumProd(int axis, ai.djl.ndarray.types.DataType dataType)
    ai.djl.ndarray.NDArray
    cumSum()
    ai.djl.ndarray.NDArray
    cumSum(int axis)
    void
    detach()
    ai.djl.ndarray.NDArray
    diagonal()
    ai.djl.ndarray.NDArray
    diagonal(int offset)
    ai.djl.ndarray.NDArray
    diagonal(int offset, int axis1, int axis2)
    ai.djl.ndarray.NDArray
    diff(int n, int dim)
    ai.djl.ndarray.NDArray
    div(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    div(Number n)
    ai.djl.ndarray.NDArray
    divi(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    divi(Number n)
    ai.djl.ndarray.NDArray
    dot(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    duplicate()
    ai.djl.ndarray.NDArray
    eq(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    eq(Number n)
    boolean
    equals(Object obj)
    ai.djl.ndarray.NDArray
    erf()
    ai.djl.ndarray.NDArray
    erfinv()
    ai.djl.ndarray.NDArray
    exp()
    ai.djl.ndarray.NDArray
    expandDims(int axis)
    ai.djl.ndarray.NDArray
    fft(long length, long axis)
    ai.djl.ndarray.NDArray
    fft2(long[] sizes, long[] axes)
    ai.djl.ndarray.NDArray
    flatten()
    ai.djl.ndarray.NDArray
    flatten(int startDim, int endDim)
    ai.djl.ndarray.NDArray
    flip(int... axes)
    ai.djl.ndarray.NDArray
    floor()
    ai.djl.ndarray.NDArray
    gammaln()
    ai.djl.ndarray.NDArray
    gather(ai.djl.ndarray.NDArray index, int axis)
    ai.djl.ndarray.NDArray
    gatherNd(ai.djl.ndarray.NDArray index)
    ai.djl.ndarray.types.DataType
    getDataType()
    ai.djl.Device
    getDevice()
    ai.djl.ndarray.NDArray
    getGradient()
    TfNDManager
    getManager()
    String
    getName()
    ai.djl.ndarray.internal.NDArrayEx
    getNDArrayInternal()
    ai.djl.ndarray.types.Shape
    getShape()
    ai.djl.ndarray.types.SparseFormat
    getSparseFormat()
    ai.djl.ndarray.NDArray
    gt(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    gt(Number n)
    ai.djl.ndarray.NDArray
    gte(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    gte(Number n)
    boolean
    hasGradient()
    int
    hashCode()
    ai.djl.ndarray.NDArray
    ifft(long length, long axis)
    ai.djl.ndarray.NDArray
    ifft2(long[] sizes, long[] axes)
    void
    intern(ai.djl.ndarray.NDArray replaced)
    ai.djl.ndarray.NDArray
    inverse()
    ai.djl.ndarray.NDArray
    irfft(long length, long axis)
    ai.djl.ndarray.NDArray
    isInfinite()
    ai.djl.ndarray.NDArray
    isNaN()
    ai.djl.ndarray.NDArray
    log()
    ai.djl.ndarray.NDArray
    log10()
    ai.djl.ndarray.NDArray
    log2()
    ai.djl.ndarray.NDArray
    logicalAnd(ai.djl.ndarray.NDArray n)
    ai.djl.ndarray.NDArray
    logicalNot()
    ai.djl.ndarray.NDArray
    logicalOr(ai.djl.ndarray.NDArray n)
    ai.djl.ndarray.NDArray
    logicalXor(ai.djl.ndarray.NDArray n)
    ai.djl.ndarray.NDArray
    logSoftmax(int axis)
    ai.djl.ndarray.NDArray
    lt(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    lt(Number n)
    ai.djl.ndarray.NDArray
    lte(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    lte(Number n)
    ai.djl.ndarray.NDArray
    matMul(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    max()
    ai.djl.ndarray.NDArray
    max(int[] axes, boolean keepDims)
    ai.djl.ndarray.NDArray
    maximum(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    maximum(Number n)
    ai.djl.ndarray.NDArray
    mean()
    ai.djl.ndarray.NDArray
    mean(int[] axes, boolean keepDims)
    ai.djl.ndarray.NDArray
    median()
    ai.djl.ndarray.NDArray
    median(int[] axes)
    ai.djl.ndarray.NDArray
    min()
    ai.djl.ndarray.NDArray
    min(int[] axes, boolean keepDims)
    ai.djl.ndarray.NDArray
    minimum(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    minimum(Number n)
    ai.djl.ndarray.NDArray
    mod(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    mod(Number n)
    ai.djl.ndarray.NDArray
    modi(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    modi(Number n)
    ai.djl.ndarray.NDArray
    mul(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    mul(Number n)
    ai.djl.ndarray.NDArray
    muli(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    muli(Number n)
    ai.djl.ndarray.NDArray
    neg()
    ai.djl.ndarray.NDArray
    negi()
    ai.djl.ndarray.NDArray
    neq(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    neq(Number n)
    ai.djl.ndarray.NDArray
    nonzero()
    ai.djl.ndarray.NDArray
    norm(boolean keepDims)
    ai.djl.ndarray.NDArray
    norm(int ord, int[] axes, boolean keepDims)
    ai.djl.ndarray.NDArray
    normalize(double p, long dim, double eps)
    ai.djl.ndarray.NDArray
    oneHot(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)
    ai.djl.ndarray.NDArray
    pad(ai.djl.ndarray.types.Shape padding, double value)
    ai.djl.ndarray.NDArray
    percentile(Number percentile)
    ai.djl.ndarray.NDArray
    percentile(Number percentile, int[] dimension)
    ai.djl.ndarray.NDArray
    pow(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    pow(Number n)
    ai.djl.ndarray.NDArray
    powi(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    powi(Number n)
    ai.djl.ndarray.NDArray
    prod()
    ai.djl.ndarray.NDArray
    prod(int[] axes, boolean keepDims)
    ai.djl.ndarray.NDArray
    put(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value)
    ai.djl.ndarray.NDArray
    real()
    ai.djl.ndarray.NDArray
    repeat(int axis, long repeats)
    ai.djl.ndarray.NDArray
    repeat(long repeats)
    ai.djl.ndarray.NDArray
    repeat(long[] repeats)
    ai.djl.ndarray.NDArray
    repeat(ai.djl.ndarray.types.Shape desiredShape)
    ai.djl.ndarray.NDArray
    reshape(ai.djl.ndarray.types.Shape shape)
    void
    returnResource(ai.djl.ndarray.NDManager manager)
    ai.djl.ndarray.NDArray
    rfft(long length, long axis)
    ai.djl.ndarray.NDArray
    rotate90(int times, int[] axes)
    ai.djl.ndarray.NDArray
    round()
    ai.djl.ndarray.NDArray
    scatter(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value, int axis)
    ai.djl.ndarray.NDArray
    sequenceMask(ai.djl.ndarray.NDArray sequenceLength)
    ai.djl.ndarray.NDArray
    sequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)
    void
    set(Buffer buffer)
    protected void
    setHandle(org.tensorflow.internal.c_api.TFE_TensorHandle newHandle)
     
    void
    setName(String name)
    void
    setRequiresGradient(boolean requiresGrad)
    ai.djl.ndarray.NDArray
    sign()
    ai.djl.ndarray.NDArray
    signi()
    ai.djl.ndarray.NDArray
    sin()
    ai.djl.ndarray.NDArray
    sinh()
    ai.djl.ndarray.NDArray
    softmax(int axis)
    ai.djl.ndarray.NDArray
    sort()
    ai.djl.ndarray.NDArray
    sort(int axis)
    ai.djl.ndarray.NDList
    split(long[] indices, int axis)
    ai.djl.ndarray.NDArray
    sqrt()
    ai.djl.ndarray.NDArray
    square()
    ai.djl.ndarray.NDArray
    squeeze(int[] axes)
    ai.djl.ndarray.NDArray
    stft(long nFft, long hopLength, boolean center, ai.djl.ndarray.NDArray window, boolean normalize, boolean returnComplex)
    ai.djl.ndarray.NDArray
    stopGradient()
    ai.djl.ndarray.NDArray
    sub(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    sub(Number n)
    ai.djl.ndarray.NDArray
    subi(ai.djl.ndarray.NDArray other)
    ai.djl.ndarray.NDArray
    subi(Number n)
    ai.djl.ndarray.NDArray
    sum()
    ai.djl.ndarray.NDArray
    sum(int[] axes, boolean keepDims)
    ai.djl.ndarray.NDArray
    take(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)
    ai.djl.ndarray.NDArray
    tan()
    ai.djl.ndarray.NDArray
    tanh()
    void
    tempAttach(ai.djl.ndarray.NDManager manager)
    ai.djl.ndarray.NDArray
    tile(int axis, long repeats)
    ai.djl.ndarray.NDArray
    tile(long repeats)
    ai.djl.ndarray.NDArray
    tile(long[] repeats)
    ai.djl.ndarray.NDArray
    tile(ai.djl.ndarray.types.Shape desiredShape)
    ByteBuffer
    toByteBuffer(boolean tryDirect)
    ai.djl.ndarray.NDArray
    toDegrees()
    ai.djl.ndarray.NDArray
    toDense()
    ai.djl.ndarray.NDArray
    toDevice(ai.djl.Device device, boolean copy)
    ai.djl.ndarray.NDList
    topK(int k, int axis, boolean largest, boolean sorted)
    ai.djl.ndarray.NDArray
    toRadians()
    ai.djl.ndarray.NDArray
    toSparse(ai.djl.ndarray.types.SparseFormat fmt)
    String
    toString()
    String[]
    toStringArray(Charset charset)
    ai.djl.ndarray.NDArray
    toType(ai.djl.ndarray.types.DataType dataType, boolean copy)
    ai.djl.ndarray.NDArray
    trace(int offset, int axis1, int axis2)
    ai.djl.ndarray.NDArray
    transpose()
    ai.djl.ndarray.NDArray
    transpose(int... dimensions)
    ai.djl.ndarray.NDArray
    trunc()
    ai.djl.ndarray.NDList
    unique(Integer dim, boolean sorted, boolean returnInverse, boolean returnCounts)
    ai.djl.ndarray.NDArray
    xlogy(ai.djl.ndarray.NDArray other)
     

    Methods inherited from class ai.djl.util.NativeResource

    getHandle, getUid, isReleased, onClose

    Methods inherited from class java.lang.Object

    clone, finalize, getClass, notify, notifyAll, wait, wait, wait

    Methods inherited from interface ai.djl.ndarray.BytesSupplier

    getAsBytes, getAsObject, getAsString

    Methods inherited from interface ai.djl.ndarray.NDArray

    allClose, allClose, argSort, argSort, booleanMask, broadcast, concat, concat, copyTo, countNonzero, countNonzero, encode, fft, fft2, get, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getResourceNDArrays, getScalar, getUid, getUint8, ifft, ifft2, irfft, isEmpty, isReleased, isScalar, isSparse, like, max, mean, min, none, norm, norm, norm, normalize, normalize, oneHot, oneHot, onesLike, prod, reshape, rfft, scaleGradient, set, set, set, set, set, set, set, set, set, setScalar, shapeEquals, size, size, split, split, split, squeeze, squeeze, stack, stack, stft, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, toByteBuffer, toDebugString, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, topK, toShortArray, toStringArray, toUint8Array, toUnsignedIntArray, toUnsignedShortArray, trace, trace, unique, unique, zerosLike

  • Method Details

    • getManager

      public TfNDManager getManager()
      Specified by:
      getManager in interface ai.djl.ndarray.NDResource

    • getName

      public String getName()
      Specified by:
      getName in interface ai.djl.ndarray.NDArray

    • setName

      public void setName(String name)
      Specified by:
      setName in interface ai.djl.ndarray.NDArray

    • getDataType

      public ai.djl.ndarray.types.DataType getDataType()
      Specified by:
      getDataType in interface ai.djl.ndarray.NDArray

    • getDevice

      public ai.djl.Device getDevice()
      Specified by:
      getDevice in interface ai.djl.ndarray.NDArray

    • getShape

      public ai.djl.ndarray.types.Shape getShape()
      Specified by:
      getShape in interface ai.djl.ndarray.NDArray

    • getSparseFormat

      public ai.djl.ndarray.types.SparseFormat getSparseFormat()
      Specified by:
      getSparseFormat in interface ai.djl.ndarray.NDArray

    • toDevice

      public ai.djl.ndarray.NDArray toDevice(ai.djl.Device device, boolean copy)
      Specified by:
      toDevice in interface ai.djl.ndarray.NDArray

    • toType

      public ai.djl.ndarray.NDArray toType(ai.djl.ndarray.types.DataType dataType, boolean copy)
      Specified by:
      toType in interface ai.djl.ndarray.NDArray

    • setRequiresGradient

      public void setRequiresGradient(boolean requiresGrad)
      Specified by:
      setRequiresGradient in interface ai.djl.ndarray.NDArray

    • getGradient

      public ai.djl.ndarray.NDArray getGradient()
      Specified by:
      getGradient in interface ai.djl.ndarray.NDArray

    • hasGradient

      public boolean hasGradient()
      Specified by:
      hasGradient in interface ai.djl.ndarray.NDArray

    • stopGradient

      public ai.djl.ndarray.NDArray stopGradient()
      Specified by:
      stopGradient in interface ai.djl.ndarray.NDArray

    • toStringArray

      public String[] toStringArray(Charset charset)
      Specified by:
      toStringArray in interface ai.djl.ndarray.NDArray

    • toByteBuffer

      public ByteBuffer toByteBuffer(boolean tryDirect)
      Specified by:
      toByteBuffer in interface ai.djl.ndarray.NDArray

    • set

      public void set(Buffer buffer)
      Specified by:
      set in interface ai.djl.ndarray.NDArray

    • gather

      public ai.djl.ndarray.NDArray gather(ai.djl.ndarray.NDArray index, int axis)
      Specified by:
      gather in interface ai.djl.ndarray.NDArray

    • gatherNd

      public ai.djl.ndarray.NDArray gatherNd(ai.djl.ndarray.NDArray index)
      Specified by:
      gatherNd in interface ai.djl.ndarray.NDArray

    • take

      public ai.djl.ndarray.NDArray take(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)
      Specified by:
      take in interface ai.djl.ndarray.NDArray

    • put

      public ai.djl.ndarray.NDArray put(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value)
      Specified by:
      put in interface ai.djl.ndarray.NDArray

    • scatter

      public ai.djl.ndarray.NDArray scatter(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value, int axis)
      Specified by:
      scatter in interface ai.djl.ndarray.NDArray

    • attach

      public void attach(ai.djl.ndarray.NDManager manager)
      Specified by:
      attach in interface ai.djl.ndarray.NDResource

    • returnResource

      public void returnResource(ai.djl.ndarray.NDManager manager)
      Specified by:
      returnResource in interface ai.djl.ndarray.NDResource

    • tempAttach

      public void tempAttach(ai.djl.ndarray.NDManager manager)
      Specified by:
      tempAttach in interface ai.djl.ndarray.NDResource

    • detach

      public void detach()
      Specified by:
      detach in interface ai.djl.ndarray.NDResource

    • duplicate

      public ai.djl.ndarray.NDArray duplicate()
      Specified by:
      duplicate in interface ai.djl.ndarray.NDArray

    • booleanMask

      public ai.djl.ndarray.NDArray booleanMask(ai.djl.ndarray.NDArray index, int axis)
      Specified by:
      booleanMask in interface ai.djl.ndarray.NDArray

    • sequenceMask

      public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)
      Specified by:
      sequenceMask in interface ai.djl.ndarray.NDArray

    • sequenceMask

      public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength)
      Specified by:
      sequenceMask in interface ai.djl.ndarray.NDArray

    • contentEquals

      public boolean contentEquals(Number number)
      Specified by:
      contentEquals in interface ai.djl.ndarray.NDArray

    • contentEquals

      public boolean contentEquals(ai.djl.ndarray.NDArray other)
      Specified by:
      contentEquals in interface ai.djl.ndarray.NDArray

    • eq

      public ai.djl.ndarray.NDArray eq(Number n)
      Specified by:
      eq in interface ai.djl.ndarray.NDArray

    • eq

      public ai.djl.ndarray.NDArray eq(ai.djl.ndarray.NDArray other)
      Specified by:
      eq in interface ai.djl.ndarray.NDArray

    • neq

      public ai.djl.ndarray.NDArray neq(Number n)
      Specified by:
      neq in interface ai.djl.ndarray.NDArray

    • neq

      public ai.djl.ndarray.NDArray neq(ai.djl.ndarray.NDArray other)
      Specified by:
      neq in interface ai.djl.ndarray.NDArray

    • gt

      public ai.djl.ndarray.NDArray gt(Number n)
      Specified by:
      gt in interface ai.djl.ndarray.NDArray

    • gt

      public ai.djl.ndarray.NDArray gt(ai.djl.ndarray.NDArray other)
      Specified by:
      gt in interface ai.djl.ndarray.NDArray

    • gte

      public ai.djl.ndarray.NDArray gte(Number n)
      Specified by:
      gte in interface ai.djl.ndarray.NDArray

    • gte

      public ai.djl.ndarray.NDArray gte(ai.djl.ndarray.NDArray other)
      Specified by:
      gte in interface ai.djl.ndarray.NDArray

    • lt

      public ai.djl.ndarray.NDArray lt(Number n)
      Specified by:
      lt in interface ai.djl.ndarray.NDArray

    • lt

      public ai.djl.ndarray.NDArray lt(ai.djl.ndarray.NDArray other)
      Specified by:
      lt in interface ai.djl.ndarray.NDArray

    • lte

      public ai.djl.ndarray.NDArray lte(Number n)
      Specified by:
      lte in interface ai.djl.ndarray.NDArray

    • lte

      public ai.djl.ndarray.NDArray lte(ai.djl.ndarray.NDArray other)
      Specified by:
      lte in interface ai.djl.ndarray.NDArray

    • all

      public ai.djl.ndarray.NDArray all()
      Specified by:
      all in interface ai.djl.ndarray.NDArray

    • any

      public ai.djl.ndarray.NDArray any()
      Specified by:
      any in interface ai.djl.ndarray.NDArray

    • erfinv

      public ai.djl.ndarray.NDArray erfinv()
      Specified by:
      erfinv in interface ai.djl.ndarray.NDArray

    • erf

      public ai.djl.ndarray.NDArray erf()
      Specified by:
      erf in interface ai.djl.ndarray.NDArray

    • norm

      public ai.djl.ndarray.NDArray norm(boolean keepDims)
      Specified by:
      norm in interface ai.djl.ndarray.NDArray

    • norm

      public ai.djl.ndarray.NDArray norm(int ord, int[] axes, boolean keepDims)
      Specified by:
      norm in interface ai.djl.ndarray.NDArray

    • oneHot

      public ai.djl.ndarray.NDArray oneHot(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)
      Specified by:
      oneHot in interface ai.djl.ndarray.NDArray

    • batchDot

      public ai.djl.ndarray.NDArray batchDot(ai.djl.ndarray.NDArray other)
      Specified by:
      batchDot in interface ai.djl.ndarray.NDArray

    • complex

      public ai.djl.ndarray.NDArray complex()
      Specified by:
      complex in interface ai.djl.ndarray.NDArray

    • real

      public ai.djl.ndarray.NDArray real()
      Specified by:
      real in interface ai.djl.ndarray.NDArray

    • conj

      public ai.djl.ndarray.NDArray conj()
      Specified by:
      conj in interface ai.djl.ndarray.NDArray

    • add

      public ai.djl.ndarray.NDArray add(Number n)
      Specified by:
      add in interface ai.djl.ndarray.NDArray

    • add

      public ai.djl.ndarray.NDArray add(ai.djl.ndarray.NDArray other)
      Specified by:
      add in interface ai.djl.ndarray.NDArray

    • sub

      public ai.djl.ndarray.NDArray sub(Number n)
      Specified by:
      sub in interface ai.djl.ndarray.NDArray

    • sub

      public ai.djl.ndarray.NDArray sub(ai.djl.ndarray.NDArray other)
      Specified by:
      sub in interface ai.djl.ndarray.NDArray

    • mul

      public ai.djl.ndarray.NDArray mul(Number n)
      Specified by:
      mul in interface ai.djl.ndarray.NDArray

    • mul

      public ai.djl.ndarray.NDArray mul(ai.djl.ndarray.NDArray other)
      Specified by:
      mul in interface ai.djl.ndarray.NDArray

    • div

      public ai.djl.ndarray.NDArray div(Number n)
      Specified by:
      div in interface ai.djl.ndarray.NDArray

    • div

      public ai.djl.ndarray.NDArray div(ai.djl.ndarray.NDArray other)
      Specified by:
      div in interface ai.djl.ndarray.NDArray

    • mod

      public ai.djl.ndarray.NDArray mod(Number n)
      Specified by:
      mod in interface ai.djl.ndarray.NDArray

    • mod

      public ai.djl.ndarray.NDArray mod(ai.djl.ndarray.NDArray other)
      Specified by:
      mod in interface ai.djl.ndarray.NDArray

    • pow

      public ai.djl.ndarray.NDArray pow(Number n)
      Specified by:
      pow in interface ai.djl.ndarray.NDArray

    • pow

      public ai.djl.ndarray.NDArray pow(ai.djl.ndarray.NDArray other)
      Specified by:
      pow in interface ai.djl.ndarray.NDArray

    • maximum

      public ai.djl.ndarray.NDArray maximum(Number n)
      Specified by:
      maximum in interface ai.djl.ndarray.NDArray

    • maximum

      public ai.djl.ndarray.NDArray maximum(ai.djl.ndarray.NDArray other)
      Specified by:
      maximum in interface ai.djl.ndarray.NDArray

    • minimum

      public ai.djl.ndarray.NDArray minimum(Number n)
      Specified by:
      minimum in interface ai.djl.ndarray.NDArray

    • minimum

      public ai.djl.ndarray.NDArray minimum(ai.djl.ndarray.NDArray other)
      Specified by:
      minimum in interface ai.djl.ndarray.NDArray

    • addi

      public ai.djl.ndarray.NDArray addi(Number n)
      Specified by:
      addi in interface ai.djl.ndarray.NDArray

    • addi

      public ai.djl.ndarray.NDArray addi(ai.djl.ndarray.NDArray other)
      Specified by:
      addi in interface ai.djl.ndarray.NDArray

    • subi

      public ai.djl.ndarray.NDArray subi(Number n)
      Specified by:
      subi in interface ai.djl.ndarray.NDArray

    • subi

      public ai.djl.ndarray.NDArray subi(ai.djl.ndarray.NDArray other)
      Specified by:
      subi in interface ai.djl.ndarray.NDArray

    • muli

      public ai.djl.ndarray.NDArray muli(Number n)
      Specified by:
      muli in interface ai.djl.ndarray.NDArray

    • muli

      public ai.djl.ndarray.NDArray muli(ai.djl.ndarray.NDArray other)
      Specified by:
      muli in interface ai.djl.ndarray.NDArray

    • divi

      public ai.djl.ndarray.NDArray divi(Number n)
      Specified by:
      divi in interface ai.djl.ndarray.NDArray

    • divi

      public ai.djl.ndarray.NDArray divi(ai.djl.ndarray.NDArray other)
      Specified by:
      divi in interface ai.djl.ndarray.NDArray

    • toSparse

      public ai.djl.ndarray.NDArray toSparse(ai.djl.ndarray.types.SparseFormat fmt)
      Specified by:
      toSparse in interface ai.djl.ndarray.NDArray

    • modi

      public ai.djl.ndarray.NDArray modi(Number n)
      Specified by:
      modi in interface ai.djl.ndarray.NDArray

    • modi

      public ai.djl.ndarray.NDArray modi(ai.djl.ndarray.NDArray other)
      Specified by:
      modi in interface ai.djl.ndarray.NDArray

    • powi

      public ai.djl.ndarray.NDArray powi(Number n)
      Specified by:
      powi in interface ai.djl.ndarray.NDArray

    • powi

      public ai.djl.ndarray.NDArray powi(ai.djl.ndarray.NDArray other)
      Specified by:
      powi in interface ai.djl.ndarray.NDArray

    • sign

      public ai.djl.ndarray.NDArray sign()
      Specified by:
      sign in interface ai.djl.ndarray.NDArray

    • signi

      public ai.djl.ndarray.NDArray signi()
      Specified by:
      signi in interface ai.djl.ndarray.NDArray

    • neg

      public ai.djl.ndarray.NDArray neg()
      Specified by:
      neg in interface ai.djl.ndarray.NDArray

    • negi

      public ai.djl.ndarray.NDArray negi()
      Specified by:
      negi in interface ai.djl.ndarray.NDArray

    • abs

      public ai.djl.ndarray.NDArray abs()
      Specified by:
      abs in interface ai.djl.ndarray.NDArray

    • square

      public ai.djl.ndarray.NDArray square()
      Specified by:
      square in interface ai.djl.ndarray.NDArray

    • sqrt

      public ai.djl.ndarray.NDArray sqrt()
      Specified by:
      sqrt in interface ai.djl.ndarray.NDArray

    • cbrt

      public ai.djl.ndarray.NDArray cbrt()
      Specified by:
      cbrt in interface ai.djl.ndarray.NDArray

    • floor

      public ai.djl.ndarray.NDArray floor()
      Specified by:
      floor in interface ai.djl.ndarray.NDArray

    • ceil

      public ai.djl.ndarray.NDArray ceil()
      Specified by:
      ceil in interface ai.djl.ndarray.NDArray

    • round

      public ai.djl.ndarray.NDArray round()
      Specified by:
      round in interface ai.djl.ndarray.NDArray

    • trunc

      public ai.djl.ndarray.NDArray trunc()
      Specified by:
      trunc in interface ai.djl.ndarray.NDArray

    • exp

      public ai.djl.ndarray.NDArray exp()
      Specified by:
      exp in interface ai.djl.ndarray.NDArray

    • gammaln

      public ai.djl.ndarray.NDArray gammaln()
      Specified by:
      gammaln in interface ai.djl.ndarray.NDArray

    • log

      public ai.djl.ndarray.NDArray log()
      Specified by:
      log in interface ai.djl.ndarray.NDArray

    • log10

      public ai.djl.ndarray.NDArray log10()
      Specified by:
      log10 in interface ai.djl.ndarray.NDArray

    • log2

      public ai.djl.ndarray.NDArray log2()
      Specified by:
      log2 in interface ai.djl.ndarray.NDArray

    • sin

      public ai.djl.ndarray.NDArray sin()
      Specified by:
      sin in interface ai.djl.ndarray.NDArray

    • cos

      public ai.djl.ndarray.NDArray cos()
      Specified by:
      cos in interface ai.djl.ndarray.NDArray

    • tan

      public ai.djl.ndarray.NDArray tan()
      Specified by:
      tan in interface ai.djl.ndarray.NDArray

    • asin

      public ai.djl.ndarray.NDArray asin()
      Specified by:
      asin in interface ai.djl.ndarray.NDArray

    • acos

      public ai.djl.ndarray.NDArray acos()
      Specified by:
      acos in interface ai.djl.ndarray.NDArray

    • atan

      public ai.djl.ndarray.NDArray atan()
      Specified by:
      atan in interface ai.djl.ndarray.NDArray

    • atan2

      public ai.djl.ndarray.NDArray atan2(ai.djl.ndarray.NDArray other)
      Specified by:
      atan2 in interface ai.djl.ndarray.NDArray

    • sinh

      public ai.djl.ndarray.NDArray sinh()
      Specified by:
      sinh in interface ai.djl.ndarray.NDArray

    • cosh

      public ai.djl.ndarray.NDArray cosh()
      Specified by:
      cosh in interface ai.djl.ndarray.NDArray

    • tanh

      public ai.djl.ndarray.NDArray tanh()
      Specified by:
      tanh in interface ai.djl.ndarray.NDArray

    • asinh

      public ai.djl.ndarray.NDArray asinh()
      Specified by:
      asinh in interface ai.djl.ndarray.NDArray

    • acosh

      public ai.djl.ndarray.NDArray acosh()
      Specified by:
      acosh in interface ai.djl.ndarray.NDArray

    • atanh

      public ai.djl.ndarray.NDArray atanh()
      Specified by:
      atanh in interface ai.djl.ndarray.NDArray

    • toDegrees

      public ai.djl.ndarray.NDArray toDegrees()
      Specified by:
      toDegrees in interface ai.djl.ndarray.NDArray

    • toRadians

      public ai.djl.ndarray.NDArray toRadians()
      Specified by:
      toRadians in interface ai.djl.ndarray.NDArray

    • max

      public ai.djl.ndarray.NDArray max()
      Specified by:
      max in interface ai.djl.ndarray.NDArray

    • max

      public ai.djl.ndarray.NDArray max(int[] axes, boolean keepDims)
      Specified by:
      max in interface ai.djl.ndarray.NDArray

    • min

      public ai.djl.ndarray.NDArray min()
      Specified by:
      min in interface ai.djl.ndarray.NDArray

    • min

      public ai.djl.ndarray.NDArray min(int[] axes, boolean keepDims)
      Specified by:
      min in interface ai.djl.ndarray.NDArray

    • sum

      public ai.djl.ndarray.NDArray sum()
      Specified by:
      sum in interface ai.djl.ndarray.NDArray

    • sum

      public ai.djl.ndarray.NDArray sum(int[] axes, boolean keepDims)
      Specified by:
      sum in interface ai.djl.ndarray.NDArray

    • cumProd

      public ai.djl.ndarray.NDArray cumProd(int axis)
      Specified by:
      cumProd in interface ai.djl.ndarray.NDArray

    • cumProd

      public ai.djl.ndarray.NDArray cumProd(int axis, ai.djl.ndarray.types.DataType dataType)
      Specified by:
      cumProd in interface ai.djl.ndarray.NDArray

    • prod

      public ai.djl.ndarray.NDArray prod()
      Specified by:
      prod in interface ai.djl.ndarray.NDArray

    • prod

      public ai.djl.ndarray.NDArray prod(int[] axes, boolean keepDims)
      Specified by:
      prod in interface ai.djl.ndarray.NDArray

    • mean

      public ai.djl.ndarray.NDArray mean()
      Specified by:
      mean in interface ai.djl.ndarray.NDArray

    • mean

      public ai.djl.ndarray.NDArray mean(int[] axes, boolean keepDims)
      Specified by:
      mean in interface ai.djl.ndarray.NDArray

    • normalize

      public ai.djl.ndarray.NDArray normalize(double p, long dim, double eps)
      Specified by:
      normalize in interface ai.djl.ndarray.NDArray

    • rotate90

      public ai.djl.ndarray.NDArray rotate90(int times, int[] axes)
      Specified by:
      rotate90 in interface ai.djl.ndarray.NDArray

    • trace

      public ai.djl.ndarray.NDArray trace(int offset, int axis1, int axis2)
      Specified by:
      trace in interface ai.djl.ndarray.NDArray

    • split

      public ai.djl.ndarray.NDList split(long[] indices, int axis)
      Specified by:
      split in interface ai.djl.ndarray.NDArray

    • flatten

      public ai.djl.ndarray.NDArray flatten()
      Specified by:
      flatten in interface ai.djl.ndarray.NDArray

    • flatten

      public ai.djl.ndarray.NDArray flatten(int startDim, int endDim)
      Specified by:
      flatten in interface ai.djl.ndarray.NDArray

    • fft

      public ai.djl.ndarray.NDArray fft(long length, long axis)
      Specified by:
      fft in interface ai.djl.ndarray.NDArray

    • rfft

      public ai.djl.ndarray.NDArray rfft(long length, long axis)
      Specified by:
      rfft in interface ai.djl.ndarray.NDArray

    • ifft

      public ai.djl.ndarray.NDArray ifft(long length, long axis)
      Specified by:
      ifft in interface ai.djl.ndarray.NDArray

    • irfft

      public ai.djl.ndarray.NDArray irfft(long length, long axis)
      Specified by:
      irfft in interface ai.djl.ndarray.NDArray

    • stft

      public ai.djl.ndarray.NDArray stft(long nFft, long hopLength, boolean center, ai.djl.ndarray.NDArray window, boolean normalize, boolean returnComplex)
      Specified by:
      stft in interface ai.djl.ndarray.NDArray

    • fft2

      public ai.djl.ndarray.NDArray fft2(long[] sizes, long[] axes)
      Specified by:
      fft2 in interface ai.djl.ndarray.NDArray

    • pad

      public ai.djl.ndarray.NDArray pad(ai.djl.ndarray.types.Shape padding, double value)
      Specified by:
      pad in interface ai.djl.ndarray.NDArray

    • ifft2

      public ai.djl.ndarray.NDArray ifft2(long[] sizes, long[] axes)
      Specified by:
      ifft2 in interface ai.djl.ndarray.NDArray

    • reshape

      public ai.djl.ndarray.NDArray reshape(ai.djl.ndarray.types.Shape shape)
      Specified by:
      reshape in interface ai.djl.ndarray.NDArray

    • expandDims

      public ai.djl.ndarray.NDArray expandDims(int axis)
      Specified by:
      expandDims in interface ai.djl.ndarray.NDArray

    • squeeze

      public ai.djl.ndarray.NDArray squeeze(int[] axes)
      Specified by:
      squeeze in interface ai.djl.ndarray.NDArray

    • unique

      public ai.djl.ndarray.NDList unique(Integer dim, boolean sorted, boolean returnInverse, boolean returnCounts)
      Specified by:
      unique in interface ai.djl.ndarray.NDArray

    • logicalAnd

      public ai.djl.ndarray.NDArray logicalAnd(ai.djl.ndarray.NDArray n)
      Specified by:
      logicalAnd in interface ai.djl.ndarray.NDArray

    • logicalOr

      public ai.djl.ndarray.NDArray logicalOr(ai.djl.ndarray.NDArray n)
      Specified by:
      logicalOr in interface ai.djl.ndarray.NDArray

    • logicalXor

      public ai.djl.ndarray.NDArray logicalXor(ai.djl.ndarray.NDArray n)
      Specified by:
      logicalXor in interface ai.djl.ndarray.NDArray

    • logicalNot

      public ai.djl.ndarray.NDArray logicalNot()
      Specified by:
      logicalNot in interface ai.djl.ndarray.NDArray

    • argSort

      public ai.djl.ndarray.NDArray argSort(int axis, boolean ascending)
      Specified by:
      argSort in interface ai.djl.ndarray.NDArray

    • sort

      public ai.djl.ndarray.NDArray sort(int axis)
      Specified by:
      sort in interface ai.djl.ndarray.NDArray

    • sort

      public ai.djl.ndarray.NDArray sort()
      Specified by:
      sort in interface ai.djl.ndarray.NDArray

    • softmax

      public ai.djl.ndarray.NDArray softmax(int axis)
      Specified by:
      softmax in interface ai.djl.ndarray.NDArray

    • logSoftmax

      public ai.djl.ndarray.NDArray logSoftmax(int axis)
      Specified by:
      logSoftmax in interface ai.djl.ndarray.NDArray

    • cumSum

      public ai.djl.ndarray.NDArray cumSum(int axis)
      Specified by:
      cumSum in interface ai.djl.ndarray.NDArray

    • cumSum

      public ai.djl.ndarray.NDArray cumSum()
      Specified by:
      cumSum in interface ai.djl.ndarray.NDArray

    • diagonal

      public ai.djl.ndarray.NDArray diagonal()
      Specified by:
      diagonal in interface ai.djl.ndarray.NDArray

    • diagonal

      public ai.djl.ndarray.NDArray diagonal(int offset)
      Specified by:
      diagonal in interface ai.djl.ndarray.NDArray

    • diagonal

      public ai.djl.ndarray.NDArray diagonal(int offset, int axis1, int axis2)
      Specified by:
      diagonal in interface ai.djl.ndarray.NDArray

    • intern

      public void intern(ai.djl.ndarray.NDArray replaced)
      Specified by:
      intern in interface ai.djl.ndarray.NDArray

    • isInfinite

      public ai.djl.ndarray.NDArray isInfinite()
      Specified by:
      isInfinite in interface ai.djl.ndarray.NDArray

    • inverse

      public ai.djl.ndarray.NDArray inverse()
      Specified by:
      inverse in interface ai.djl.ndarray.NDArray

    • isNaN

      public ai.djl.ndarray.NDArray isNaN()
      Specified by:
      isNaN in interface ai.djl.ndarray.NDArray

    • tile

      public ai.djl.ndarray.NDArray tile(long repeats)
      Specified by:
      tile in interface ai.djl.ndarray.NDArray

    • tile

      public ai.djl.ndarray.NDArray tile(int axis, long repeats)
      Specified by:
      tile in interface ai.djl.ndarray.NDArray

    • tile

      public ai.djl.ndarray.NDArray tile(long[] repeats)
      Specified by:
      tile in interface ai.djl.ndarray.NDArray

    • tile

      public ai.djl.ndarray.NDArray tile(ai.djl.ndarray.types.Shape desiredShape)
      Specified by:
      tile in interface ai.djl.ndarray.NDArray

    • repeat

      public ai.djl.ndarray.NDArray repeat(long repeats)
      Specified by:
      repeat in interface ai.djl.ndarray.NDArray

    • repeat

      public ai.djl.ndarray.NDArray repeat(int axis, long repeats)
      Specified by:
      repeat in interface ai.djl.ndarray.NDArray

    • repeat

      public ai.djl.ndarray.NDArray repeat(long[] repeats)
      Specified by:
      repeat in interface ai.djl.ndarray.NDArray

    • repeat

      public ai.djl.ndarray.NDArray repeat(ai.djl.ndarray.types.Shape desiredShape)
      Specified by:
      repeat in interface ai.djl.ndarray.NDArray

    • dot

      public ai.djl.ndarray.NDArray dot(ai.djl.ndarray.NDArray other)
      Specified by:
      dot in interface ai.djl.ndarray.NDArray

    • matMul

      public ai.djl.ndarray.NDArray matMul(ai.djl.ndarray.NDArray other)
      Specified by:
      matMul in interface ai.djl.ndarray.NDArray

    • batchMatMul

      public ai.djl.ndarray.NDArray batchMatMul(ai.djl.ndarray.NDArray other)
      Specified by:
      batchMatMul in interface ai.djl.ndarray.NDArray

    • xlogy

      public ai.djl.ndarray.NDArray xlogy(ai.djl.ndarray.NDArray other)
      Specified by:
      xlogy in interface ai.djl.ndarray.NDArray

    • clip

      public ai.djl.ndarray.NDArray clip(Number min, Number max)
      Specified by:
      clip in interface ai.djl.ndarray.NDArray

    • flip

      public ai.djl.ndarray.NDArray flip(int... axes)
      Specified by:
      flip in interface ai.djl.ndarray.NDArray

    • transpose

      public ai.djl.ndarray.NDArray transpose()
      Specified by:
      transpose in interface ai.djl.ndarray.NDArray

    • transpose

      public ai.djl.ndarray.NDArray transpose(int... dimensions)
      Specified by:
      transpose in interface ai.djl.ndarray.NDArray

    • broadcast

      public ai.djl.ndarray.NDArray broadcast(ai.djl.ndarray.types.Shape shape)
      Specified by:
      broadcast in interface ai.djl.ndarray.NDArray

    • argMax

      public ai.djl.ndarray.NDArray argMax()
      Specified by:
      argMax in interface ai.djl.ndarray.NDArray

    • argMax

      public ai.djl.ndarray.NDArray argMax(int axis)
      Specified by:
      argMax in interface ai.djl.ndarray.NDArray

    • topK

      public ai.djl.ndarray.NDList topK(int k, int axis, boolean largest, boolean sorted)
      Specified by:
      topK in interface ai.djl.ndarray.NDArray

    • argMin

      public ai.djl.ndarray.NDArray argMin()
      Specified by:
      argMin in interface ai.djl.ndarray.NDArray

    • argMin

      public ai.djl.ndarray.NDArray argMin(int axis)
      Specified by:
      argMin in interface ai.djl.ndarray.NDArray

    • percentile

      public ai.djl.ndarray.NDArray percentile(Number percentile)
      Specified by:
      percentile in interface ai.djl.ndarray.NDArray

    • percentile

      public ai.djl.ndarray.NDArray percentile(Number percentile, int[] dimension)
      Specified by:
      percentile in interface ai.djl.ndarray.NDArray

    • median

      public ai.djl.ndarray.NDArray median()
      Specified by:
      median in interface ai.djl.ndarray.NDArray

    • median

      public ai.djl.ndarray.NDArray median(int[] axes)
      Specified by:
      median in interface ai.djl.ndarray.NDArray

    • toDense

      public ai.djl.ndarray.NDArray toDense()
      Specified by:
      toDense in interface ai.djl.ndarray.NDArray

    • nonzero

      public ai.djl.ndarray.NDArray nonzero()
      Specified by:
      nonzero in interface ai.djl.ndarray.NDArray

    • getNDArrayInternal

      public ai.djl.ndarray.internal.NDArrayEx getNDArrayInternal()
      Specified by:
      getNDArrayInternal in interface ai.djl.ndarray.NDArray

    • diff

      public ai.djl.ndarray.NDArray diff(int n, int dim)
      Specified by:
      diff in interface ai.djl.ndarray.NDArray

    • equals

      public boolean equals(Object obj)
      Overrides:
      equals in class Object

    • hashCode

      public int hashCode()
      Overrides:
      hashCode in class Object

    • toString

      public String toString()
      Overrides:
      toString in class Object

    • close

      public void close()
      Specified by:
      close in interface AutoCloseable
      Specified by:
      close in interface ai.djl.ndarray.NDArray
      Specified by:
      close in interface ai.djl.ndarray.NDResource
      Overrides:
      close in class ai.djl.util.NativeResource<org.tensorflow.internal.c_api.TFE_TensorHandle>

    • setHandle

      protected void setHandle(org.tensorflow.internal.c_api.TFE_TensorHandle newHandle)