Package org.tensorflow.op.core

Class Shape<U extends java.lang.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 java.lang.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	Description
Output<U>	asOutput()	Returns the symbolic handle of a tensor.
static <T> Shape<java.lang.Integer>	create(Scope scope, Operand<T> input)	Factory method to create a class to wrap a new Shape operation to the graph, using default output types.
static <U extends java.lang.Number,T> Shape<U>	create(Scope scope, Operand<T> input, java.lang.Class<U> outType)	Factory method to create a class to wrap a new Shape operation to the graph.
Output<U>	output()

Methods inherited from class java.lang.Object

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

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Method Detail

create

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

Factory method to create a class to wrap a new Shape operation to the graph.

Parameters:

scope - current graph scope

input -

outType -

Returns:

a new instance of Shape

create

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

Factory method to create a class to wrap a new Shape operation to the graph, using default output types.

Parameters:

scope - current graph 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 java.lang.Number>

See Also:

OperationBuilder.addInput(Output)