Package org.tensorflow.op.core

Class Size<U extends java.lang.Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

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

Type Parameters:

U - data type for output() output

All Implemented Interfaces:

Op, Operand<U>

@Operator
public final class Size<U extends java.lang.Number>
extends PrimitiveOp
implements Operand<U>

Returns the size of a tensor.

This operation returns an integer representing the number of elements in `input`.

For example:

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

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

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

Parameters:

scope - current graph scope

input -

outType -

Returns:

a new instance of Size

create

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

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

Parameters:

scope - current graph scope

input -

Returns:

a new instance of Size

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)