Package aten
Class Tensor
java.lang.Object
aten.Tensor
-
Method Summary
Modifier and TypeMethodDescriptionvoid
add_
(double other, double alpha) void
add_l_
(long other, long alpha) static void
addmm_out_transposed1
(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha) static void
addmm_out_transposed2
(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha) static void
baddbmm_out_transposed1
(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha) static void
baddbmm_out_transposed2
(Tensor out, Tensor self, Tensor mat1, Tensor mat2, double beta, double alpha) coalesce()
void
boolean
copyFromByteArray
(byte[] data) boolean
copyFromByteArrayAtOffset
(byte[] data, long offset) boolean
copyFromDoubleArray
(double[] data) boolean
copyFromDoubleArrayAtOffset
(double[] data, long offset) boolean
copyFromFloatArray
(float[] data) boolean
copyFromFloatArrayAtOffset
(float[] data, long offset) boolean
copyFromLongArray
(long[] data) boolean
copyFromLongArrayAtOffset
(long[] data, long offset) boolean
copyToByteArray
(byte[] data) boolean
copyToDoubleArray
(double[] data) boolean
copyToFloatArray
(float[] data) boolean
copyToLongArray
(long[] data) cpu()
cuda()
static void
cudaCachingAllocatorSetMemoryFraction
(double fraction, int device) boolean
defined()
long
dim()
long
protected static Tensor
factory
(long cPtr) static Tensor
protected static long
static long
static boolean
hasCuda()
static boolean
hasMps()
indices()
boolean
boolean
isCuda()
boolean
isMps()
static long
lowlevelfrom_file
(String path, long offset, long len, byte tpe, boolean pin) static long
lowlevelones_like
(long tensor) static long[]
lowleveltensorsfrom_file
(String path, long offset, long len, boolean pin, byte[] scalarTypes, long[] tensorOffsets, long[] tensorLengths, int numTensors) static long
lowlevelzeros_like
(long tensor) static void
manual_seed
(long seed) static void
manual_seed_cpu
(long seed) static void
manual_seed_cuda
(long seed, int device) static void
manual_seed_mps
(long seed) void
mul_
(double d) void
mul_l_
(long d) long
numel()
static Tensor
options()
void
print()
void
release()
static void
releaseAll
(Tensor[] tensors) repeat
(long[] repeats) static Tensor
scalarDouble
(double scalar, TensorOptions options) static Tensor
scalarFloat
(float scalar, TensorOptions options) static Tensor
scalarLong
(long scalar, TensorOptions options) byte
byte
static void
Sets the global cpu allocator to the default allocatorstatic void
Sets the global cpu allocator to the pinned memory allocator This is needs the CUDA librarieslong[]
sizes()
long[]
strides()
static Tensor[]
tensors_from_file
(String path, long offset, long len, boolean pin, byte[] scalarTypes, long[] tensorOffsets, long[] tensorLengths) to
(TensorOptions op, boolean non_blocking, boolean copy) to_dense()
toString()
long
useCount()
values()
long
static Tensor
zeros_like
(Tensor tensor)
-
Method Details
-
factory
-
getCPtr
-
dim
public long dim() -
defined
public boolean defined() -
useCount
public long useCount() -
weakUseCount
public long weakUseCount() -
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() -
isMps
public boolean isMps() -
cuda
-
cpu
-
print
public void print() -
options
-
hasCuda
public static boolean hasCuda() -
getNumGPUs
public static long getNumGPUs() -
hasMps
public static boolean hasMps() -
scalarTypeByte
public byte scalarTypeByte() -
release
public void release() -
releaseAll
-
scalarDouble
-
scalarFloat
-
scalarLong
-
to
-
is_pinned
public boolean is_pinned() -
pin_memory
-
addmm_out_transposed1
-
addmm_out_transposed2
-
baddbmm_out_transposed1
-
baddbmm_out_transposed2
-
toString
-
copyFrom
-
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) -
copyFromByteArray
public boolean copyFromByteArray(byte[] data) -
copyFromByteArrayAtOffset
public boolean copyFromByteArrayAtOffset(byte[] data, long offset) -
copyToByteArray
public boolean copyToByteArray(byte[] 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
-
to_dense
-
values
-
indices
-
coalesce
-
repeat
-
from_file
-
tensors_from_file
-
manual_seed
public static void manual_seed(long seed) -
manual_seed_mps
public static void manual_seed_mps(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
-
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
-
lowlevelones_like
public static long lowlevelones_like(long tensor) -
ones_like
-
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
-