org.tensorflow.op.core

Class Shape<U extends Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.Shape<U>

Type Parameters:

U - data type for output() output

All Implemented Interfaces:

Op, Operand<U>

@Operator
public final class Shape<U extends Number>
extends PrimitiveOp
implements Operand<U>

Returns the shape of a tensor.

This operation returns a 1-D integer tensor representing the shape of `input`.

For example:

 # 't' is [[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]]
 shape(t) ==> [2, 2, 3]
 

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<U>	asOutput() Returns the symbolic handle of a tensor.
static <T> Shape<Integer>	create(Scope scope, Operand<T> input) Factory method to create a class wrapping a new Shape operation using default output types.
static <U extends Number,T> Shape<U>	create(Scope scope, Operand<T> input, Class<U> outType) Factory method to create a class wrapping a new Shape operation.
Output<U>	output()

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static <U extends Number,T> Shape<U> create(Scope scope,
                                                   Operand<T> input,
                                                   Class<U> outType)

Factory method to create a class wrapping a new Shape operation.

Parameters:

scope - current scope

input -

outType -

Returns:

a new instance of Shape

create

public static <T> Shape<Integer> create(Scope scope,
                                        Operand<T> input)

Factory method to create a class wrapping a new Shape operation using default output types.

Parameters:

scope - current scope

input -

Returns:

a new instance of Shape

output

public Output<U> output()

asOutput

public Output<U> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

Specified by:

asOutput in interface Operand<U extends Number>

See Also:

OperationBuilder.addInput(Output)