io.improbable.keanu.tensor.ndj4

Class INDArrayExtensions

java.lang.Object

io.improbable.keanu.tensor.ndj4.INDArrayExtensions

public class INDArrayExtensions
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
INDArrayExtensions()

Method Summary

Modifier and Type	Method and Description
static org.nd4j.linalg.api.ndarray.INDArray	castToInteger(org.nd4j.linalg.api.ndarray.INDArray tensor, boolean duplicate)
static org.nd4j.linalg.api.ndarray.INDArray	cumProd(org.nd4j.linalg.api.ndarray.INDArray array, int dimension) Cumulative prod along a dimension.
static java.util.List<org.nd4j.linalg.api.ndarray.INDArray>	split(org.nd4j.linalg.api.ndarray.INDArray tensor, int dimension, long... splitAtIndices)

Methods inherited from class java.lang.Object

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

Constructor Detail

INDArrayExtensions

public INDArrayExtensions()

Method Detail

castToInteger

public static org.nd4j.linalg.api.ndarray.INDArray castToInteger(org.nd4j.linalg.api.ndarray.INDArray tensor,
                                                                 boolean duplicate)

split

public static java.util.List<org.nd4j.linalg.api.ndarray.INDArray> split(org.nd4j.linalg.api.ndarray.INDArray tensor,
                                                                         int dimension,
                                                                         long... splitAtIndices)

cumProd

public static org.nd4j.linalg.api.ndarray.INDArray cumProd(org.nd4j.linalg.api.ndarray.INDArray array,
                                                           int dimension)

Cumulative prod along a dimension. This code is copied from the cumSumi implementation in org.nd4j.linalg.api.ndarray.BaseNDArray.java

Parameters:

array - array to cumProd

dimension - the dimension to perform cumulative product along

Returns:

the cumulative product along the specified dimension