org.deeplearning4j.streaming.conversion.ndarray

Class NDArrayRecordToNDArray

java.lang.Object

org.deeplearning4j.streaming.conversion.ndarray.NDArrayRecordToNDArray

All Implemented Interfaces:

java.io.Serializable, RecordToNDArray

public class NDArrayRecordToNDArray
extends java.lang.Object
implements RecordToNDArray

Assumes all records in the given batch are of type @link{NDArrayWritable} It extracts the underlying arrays and returns a concatenated array.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
NDArrayRecordToNDArray()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	convert(java.util.Collection<java.util.Collection<org.datavec.api.writable.Writable>> records) Converts a list of records in to 1 ndarray

Methods inherited from class java.lang.Object

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

Constructor Detail

NDArrayRecordToNDArray

public NDArrayRecordToNDArray()

Method Detail

convert

public org.nd4j.linalg.api.ndarray.INDArray convert(java.util.Collection<java.util.Collection<org.datavec.api.writable.Writable>> records)

Description copied from interface: RecordToNDArray

Converts a list of records in to 1 ndarray

Specified by:

convert in interface RecordToNDArray

Parameters:

records - the records to convert

Returns:

the collection of records to convert