Package aten

Class Tensor

java.lang.Object
aten.Tensor
public class Tensor extends Object

  • Method Details

    • factory

      protected static Tensor factory(long cPtr)

    • getCPtr

      protected static long getCPtr(Tensor obj)

    • dim

      public long dim()

    • defined

      public boolean defined()

    • useCount

      public long useCount()

    • weakUseCount

      public long weakUseCount()

    • nativeToString

      public String nativeToString()

    • sizes

      public long[] sizes()

    • strides

      public long[] strides()

    • numel

      public long numel()

    • elementSize

      public long elementSize()

    • scalarType

      public byte scalarType()

    • isCuda

      public boolean isCuda()

    • cuda

      public Tensor cuda()

    • cpu

      public Tensor cpu()

    • print

      public void print()

    • options

      public TensorOptions options()

    • cudnnAvailable

      public static boolean cudnnAvailable()

    • getNumGPUs

      public static long getNumGPUs()

    • scalarTypeByte

      public byte scalarTypeByte()

    • release

      public void release()

    • releaseAll

      public static void releaseAll(Tensor[] tensors)

    • scalarDouble

      public static Tensor scalarDouble(double scalar, TensorOptions options)

    • scalarFloat

      public static Tensor scalarFloat(float scalar, TensorOptions options)

    • scalarLong

      public static Tensor scalarLong(long scalar, TensorOptions options)

    • to

      public Tensor to(TensorOptions op, boolean non_blocking, boolean copy)

    • is_pinned

      public boolean is_pinned()

    • pin_memory

      public Tensor pin_memory()

    • addmm_out_transposed1

      public static void addmm_out_transposed1(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha)

    • addmm_out_transposed2

      public static void addmm_out_transposed2(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha)

    • baddbmm_out_transposed1

      public static void baddbmm_out_transposed1(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha)

    • baddbmm_out_transposed2

      public static void baddbmm_out_transposed2(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha)

    • toString

      public String toString()
      Overrides:
      toString in class Object

    • copyFrom

      public void copyFrom(Tensor other, boolean nonBlocking)

    • copyFromFloatArray

      public boolean copyFromFloatArray(float[] data)

    • copyFromFloatArrayAtOffset

      public boolean copyFromFloatArrayAtOffset(float[] data, long offset)

    • copyToFloatArray

      public boolean copyToFloatArray(float[] data)

    • copyFromDoubleArray

      public boolean copyFromDoubleArray(double[] data)

    • copyFromDoubleArrayAtOffset

      public boolean copyFromDoubleArrayAtOffset(double[] data, long offset)

    • copyToDoubleArray

      public boolean copyToDoubleArray(double[] data)

    • copyFromLongArray

      public boolean copyFromLongArray(long[] data)

    • copyFromLongArrayAtOffset

      public boolean copyFromLongArrayAtOffset(long[] data, long offset)

    • copyToLongArray

      public boolean copyToLongArray(long[] data)

    • mul_

      public void mul_(double d)

    • mul_l_

      public void mul_l_(long d)

    • add_

      public void add_(double other, double alpha)

    • add_l_

      public void add_l_(long other, long alpha)

    • expand_as

      public Tensor expand_as(Tensor other)

    • to_dense

      public Tensor to_dense()

    • values

      public Tensor values()

    • indices

      public Tensor indices()

    • coalesce

      public Tensor coalesce()

    • repeat

      public Tensor repeat(long[] repeats)

    • from_file

      public static Tensor from_file(String path, long offset, long len, byte tpe, boolean pin)

    • tensors_from_file

      public static Tensor[] tensors_from_file(String path, long offset, long len, boolean pin, byte[] scalarTypes, long[] tensorOffsets, long[] tensorLengths)

    • manual_seed

      public static void manual_seed(long seed)

    • manual_seed_cpu

      public static void manual_seed_cpu(long seed)

    • manual_seed_cuda

      public static void manual_seed_cuda(long seed, int device)

    • lowlevelfrom_file

      public static long lowlevelfrom_file(String path, long offset, long len, byte tpe, boolean pin)

    • lowleveltensorsfrom_file

      public static long[] lowleveltensorsfrom_file(String path, long offset, long len, boolean pin, byte[] scalarTypes, long[] tensorOffsets, long[] tensorLengths, int numTensors)

    • lowlevelzeros_like

      public static long lowlevelzeros_like(long tensor)

    • zeros_like

      public static Tensor zeros_like(Tensor tensor)

    • lowlevelones_like

      public static long lowlevelones_like(long tensor)

    • ones_like

      public static Tensor ones_like(Tensor tensor)

    • cudaCachingAllocatorSetMemoryFraction

      public static void cudaCachingAllocatorSetMemoryFraction(double fraction, int device)

    • setPinnedMemoryAllocator

      public static void setPinnedMemoryAllocator()
      Sets the global cpu allocator to the pinned memory allocator This is needs the CUDA libraries

    • setDefaultAllocator

      public static void setDefaultAllocator()
      Sets the global cpu allocator to the default allocator