org.nd4j.linalg.util

Class NDArrayUtil

java.lang.Object

org.nd4j.linalg.util.NDArrayUtil

public class NDArrayUtil
extends java.lang.Object

Basic INDArray ops

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	NDArrayUtil.DimensionOp
static class	NDArrayUtil.MatrixOp
static class	NDArrayUtil.ScalarOp

Constructor Summary

Constructors

Constructor and Description
NDArrayUtil()

Method Summary

Modifier and Type	Method and Description
static INDArray	center(INDArray arr, int[] shape) Center an array
static INDArray	exp(INDArray toExp)
static INDArray	expi(INDArray toExp) Returns an exponential version of this ndarray
static INDArray	padWithZeros(INDArray nd, int[] targetShape) Pads an INDArray with zeros
static INDArray	truncate(INDArray nd, int n, int dimension) Truncates an INDArray to the specified shape.

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

NDArrayUtil

public NDArrayUtil()

Method Detail

exp

public static INDArray exp(INDArray toExp)

expi

public static INDArray expi(INDArray toExp)

Returns an exponential version of this ndarray

Parameters:

toExp - the INDArray to convert

Returns:

the converted ndarray

center

public static INDArray center(INDArray arr,
                              int[] shape)

Center an array

Parameters:

arr - the arr to center

shape - the shape of the array

Returns:

the center portion of the array based on the specified shape

truncate

public static INDArray truncate(INDArray nd,
                                int n,
                                int dimension)

Truncates an INDArray to the specified shape. If the shape is the same or greater, it just returns the original array

Parameters:

nd - the INDArray to truncate

n - the number of elements to truncate to

Returns:

the truncated ndarray

padWithZeros

public static INDArray padWithZeros(INDArray nd,
                                    int[] targetShape)

Pads an INDArray with zeros

Parameters:

nd - the INDArray to pad

targetShape - the the new shape

Returns:

the padded ndarray