Have a question? Connect with the community at the TensorFlow Forum Visit Forum

Shapes

public abstract class Shapes

An operator providing methods on org.tensorflow.op.core.Shape tensors and 1d operands that represent the dimensions of a shape.

Ops tf = Ops.create();
 Operand<TFloat32> yPred = tf.constant(new float[][][] { { {0f, 0.5f, {0.3f, 0.9f} } });
 Shape predShape = tf.shape(yPred);
 Operand numLabels =
     tf.reduceProd(tf.shape.takeLast(predShape, // take all but the first dimension
         tf.math.sub(  // by subtract 1 from the total dimensions represent by  predShape
             tf.shape.numDimensions(predShape),
             tf.constant(1))),
         tf.constant(0));

 Operand numPred = tf.shape.size(predShape, tf.constant(0));
 Operand predFlat = tf.shape.flatten(yPred);

 Shape predShape64 = tf.shape(yPred, TInt64.class);
 Operand predSqueezed = tf.shape.squeeze(predShape64, TInt64.class);
 }

Public Constructors

Public Methods

static Operand < TInt64 >
append ( Scope scope, Shape < TInt64 > shape, long lastDimension)
Creates a 1-dimensional operand containing the dimensions of a shape followed by the last dimension.
static Operand < TInt32 >
append ( Scope scope, Shape < TInt32 > shape, int lastDimension)
Creates a 1-dimensional operand containing the dimensions of a shape followed by the last dimension.
static <T extends TNumber > Operand <T>
append ( Scope scope, Operand <T> shape, Operand <T> shapeToAppend)
Creates a 1-dimensional operand that represents a new shape containing the dimensions of the operand representing a shape, followed by the dimensions of an operand representing a shape to append.
static <T extends TType , U extends TNumber > Operand <T>
flatten ( Scope scope, Operand <T> operand, Class<U> type)
Flatten the operand to 1 dimension
static <U extends TNumber > Operand <U>
flatten ( Scope scope, Shape <U> shape, Class<U> type)
Flatten the shape to 1 dimension.
static <T extends TType > Operand <T>
flatten ( Scope scope, Operand <T> operand)
Flatten the operand to 1 dimension.
static Operand < TInt32 >
flatten ( Scope scope, Shape < TInt32 > shape)
Flatten the shape to 1 dimension.
static <U extends TNumber > Operand <U>
head ( Scope scope, Shape <U> shape, Class<U> type)
Creates a 1-dimensional Operand containing the Shape's first dimension.
static Operand < TInt32 >
head ( Scope scope, Shape < TInt32 > shape)
Creates a 1-dimensional Operand containing the Shape's first dimension.
static Operand < TInt32 >
numDimensions ( Scope scope, Shape < TInt32 > shape)
Get the number of dimensions of the shape object.
static <U extends TNumber > Operand <U>
numDimensions ( Scope scope, Shape <U> shape, Class<U> type)
Get the number of dimensions of the shape object.
static Operand < TInt64 >
prepend ( Scope scope, Shape < TInt64 > shape, long firstDimension)
Creates a 1-dimensional operand containing the first dimension followed by the dimensions of the shape.
static <T extends TNumber > Operand <T>
prepend ( Scope scope, Operand <T> shape, Operand <T> shapeToPrepend)
Creates a 1-dimensional operand that represents a new shape containing the dimensions of an operand representing the shape to prepend, followed by the dimensions of an operand representing a shape.
static Operand < TInt32 >
prepend ( Scope scope, Shape < TInt32 > shape, int firstDimension)
Creates a 1-dimensional operand containing the first dimension followed by the dimensions of the shape.
static <U extends TNumber > Operand <U>
reduceDims ( Scope scope, Shape <U> shape, Operand <U> axis, Class<U> type)
Reduces the shape to the specified axis.
static Operand < TInt32 >
reduceDims ( Scope scope, Shape < TInt32 > shape, Operand < TInt32 > axis)
Reduces the shape to the specified axis.
static <T extends TType > Operand <T>
reduceDims ( Scope scope, Operand <T> operand, Operand < TInt32 > axis)
Reshapes the operand by reducing the shape to the specified axis.
static <T extends TType , U extends TNumber > Operand <T>
reduceDims ( Scope scope, Operand <T> operand, Operand <U> axis, Class<U> type)
Reshapes the operand by reducing the shape to the specified axis.
static <U extends TNumber > Operand <U>
size ( Scope scope, Shape <U> shape, Class<U> type)
Get the size represented by the TensorFlow shape.
static <T extends TType > Operand < TInt32 >
size ( Scope scope, Operand <T> input, Operand < TInt32 > dim)
Get the size of the specified dimension for the shape of the tensor.
static Operand < TInt32 >
size ( Scope scope, Shape < TInt32 > shape)
Get the size represented by the TensorFlow shape.
static Operand < TInt32 >
size ( Scope scope, Shape < TInt32 > shape, Operand < TInt32 > dim)
Get the size of the specified dimension in the shape.
static <U extends TNumber > Operand <U>
size ( Scope scope, Shape <U> shape, Operand <U> dim, Class<U> type)
Get the size of the specified dimension in the shape.
static <T extends TType , U extends TNumber > Operand <U>
size ( Scope scope, Operand <T> input, Operand <U> dim, Class<U> type)
Get the size of the specified dimension for the shape of the tensor.
static Operand < TInt32 >
squeeze ( Scope scope, Shape < TInt32 > shape)
Removes dimensions of size 1 from the shape.
static <U extends TNumber > Operand <U>
squeeze ( Scope scope, Shape <U> shape, Class<U> type)
Removes dimensions of size 1 from the shape.
static Operand < TInt32 >
tail ( Scope scope, Shape < TInt32 > shape)
Creates a 1-dimensional Operand that contains the dimension matching the last dimension of the Shape.
static <U extends TNumber > Operand <U>
tail ( Scope scope, Shape <U> shape, Class<U> type)
Creates a 1-dimensional Operand that contains the dimension matching the last dimension of * the Shape.
static Operand < TInt32 >
take ( Scope scope, Shape < TInt32 > shape, Operand < TInt32 > n)
Creates a 1-dimensional operand with the dimensions matching the first n dimensions of the shape.
static <U extends TNumber > Operand <U>
take ( Scope scope, Shape <U> shape, Operand <U> n, Class<U> type)
Creates a 1-dimensional operand containin the dimensions matching the first n dimensions of the shape.
static <U extends TNumber > Operand <U>
takeLast ( Scope scope, Shape <U> shape, Operand <U> n, Class<U> type)
Creates a 1-dimensional operand containing the dimensions matching the last n dimensions of the shape.
static <U extends TNumber > Operand < TInt32 >
takeLast ( Scope scope, Shape < TInt32 > shape, Operand < TInt32 > n)
Creates a 1-dimensional operand containing the dimensions matching the last n dimensions of the shape.

Inherited Methods

Public Constructors

public Shapes ()

Public Methods

public static Operand < TInt64 > append ( Scope scope, Shape < TInt64 > shape, long lastDimension)

Creates a 1-dimensional operand containing the dimensions of a shape followed by the last dimension.

Parameters
scope current scope
shape the TensorFlow shape
lastDimension the dimension(s) to append
Returns
  • a 1-dimensional operand containing the dimensions of a shape followed by the last dimension

public static Operand < TInt32 > append ( Scope scope, Shape < TInt32 > shape, int lastDimension)

Creates a 1-dimensional operand containing the dimensions of a shape followed by the last dimension.

Parameters
scope current scope
shape the TensorFlow shape
lastDimension the dimension(s) to append
Returns
  • a 1-dimensional operand containing the dimensions of a shape followed by the last dimension

public static Operand <T> append ( Scope scope, Operand <T> shape, Operand <T> shapeToAppend)

Creates a 1-dimensional operand that represents a new shape containing the dimensions of the operand representing a shape, followed by the dimensions of an operand representing a shape to append.

Parameters
scope current scope
shape the TensorFlow shape
shapeToAppend the other shape to append
Returns
  • a 1-dimensional operand that represents a new shape containing the dimensions of the operand representing a shape, followed by the dimensions of an operand representing a shape to append

public static Operand <T> flatten ( Scope scope, Operand <T> operand, Class<U> type)

Flatten the operand to 1 dimension

Parameters
scope current scope
operand the operand to flatten
type the shape datatype
Returns
  • the reshaped operand

public static Operand <U> flatten ( Scope scope, Shape <U> shape, Class<U> type)

Flatten the shape to 1 dimension.

Parameters
scope current scope
shape the TensorFlow shape
type the shape datatype
Returns
  • the flattened shape

public static Operand <T> flatten ( Scope scope, Operand <T> operand)

Flatten the operand to 1 dimension.

Parameters
scope current scope
operand