org.nd4j.linalg.api.ops.impl.broadcast

Class BroadcastDimensions

java.lang.Object

org.nd4j.linalg.api.ops.impl.broadcast.BroadcastDimensions

public class BroadcastDimensions
extends Object

Created by agibsonccc on 11/12/15.

Constructor Summary

Constructors

Constructor and Description
BroadcastDimensions()

Method Summary

Modifier and Type	Method and Description
static int[]	getDimensions(int[] shape) Determine the valid broad cast dimensions based on which values in the given shape are equal to 1

Methods inherited from class java.lang.Object

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

Constructor Detail

BroadcastDimensions

public BroadcastDimensions()

Method Detail

getDimensions

public static int[] getDimensions(int[] shape)

Determine the valid broad cast dimensions based on which values in the given shape are equal to 1

Parameters:

shape - the shape to get the broadcast dimensions for

Returns:

the dimension indexes with a broadcast shape