Package org.tensorflow.op.math

Class Mean<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.math.Mean<T>

Type Parameters:

T - data type for output() output

All Implemented Interfaces:

Op, Operand<T>

@Operator(group="math")
public final class Mean<T>
extends PrimitiveOp
implements Operand<T>

Computes the mean of elements across dimensions of a tensor.

Reduces `input` along the dimensions given in `axis`. Unless `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in `axis`. If `keep_dims` is true, the reduced dimensions are retained with length 1.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	Mean.Options	Optional attributes for Mean

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method	Description
Output<T>	asOutput()	Returns the symbolic handle of a tensor.
static <T,U extends java.lang.Number> Mean<T>	create(Scope scope, Operand<T> input, Operand<U> axis, Mean.Options... options)	Factory method to create a class to wrap a new Mean operation to the graph.
static Mean.Options	keepDims(java.lang.Boolean keepDims)
Output<T>	output()	The reduced tensor.

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 <T,U extends java.lang.Number> Mean<T> create(Scope scope,
                                                            Operand<T> input,
                                                            Operand<U> axis,
                                                            Mean.Options... options)

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

Parameters:

scope - current graph scope

input - The tensor to reduce.

axis - The dimensions to reduce. Must be in the range `[-rank(input), rank(input))`.

options - carries optional attributes values

Returns:

a new instance of Mean

keepDims

public static Mean.Options keepDims(java.lang.Boolean keepDims)

Parameters:

keepDims - If true, retain reduced dimensions with length 1.

output

public Output<T> output()

The reduced tensor.

asOutput

public Output<T> 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<T>

See Also:

OperationBuilder.addInput(Output)