public class
ShapeUtils
Various methods for processing with Shapes and Operands
Public Constructors
Public Methods
| static int[] | |
| static <T extends TIntegral> long[] |
getLongArray(T dims)
Converts a TInt32 or TInt64 to a java long array
|
| static <T extends TIntegral> long[] | |
| static Shape | |
| static <T extends TIntegral> Shape |
Inherited Methods
Public Constructors
public ShapeUtils ()
Public Methods
public static int[] getIntArray (Scope scope, Operand<TInt32> dims)
Converts a TInt32 type Operand to a Java int array
Parameters
| scope | the TensorFlow scope |
|---|---|
| dims | the shape dimensions operand |
Returns
- the int array of the dimensions
public static long[] getLongArray (T dims)
Converts a TInt32 or TInt64 to a java long array
Parameters
| dims | the dimension tensor |
|---|
Returns
- the long array
Throws
| IllegalArgumentException | if the dims type is not an integer |
|---|
public static long[] getLongArray (Scope scope, Operand<T> dims)
Converts a TInt32 or TInt64 Operand to a java long array
Parameters
| scope | the TensorFlow scope |
|---|---|
| dims | the Operand |
Returns
- the long array
Throws
| IllegalArgumentException | if the dims type is not an integer |
|---|
public static Shape reduce (Shape shape, int axis)
Reduces the shape by eliminating trailing Dimensions.
The last dimension, specified by axis, will be a product of all remaining dimensions
Parameters
| shape | the shape to squeeze |
|---|---|
| axis | the axis to squeeze |
Returns
- the new shape