org.tensorflow.op.core

Class ShapeN<U extends Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

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

Type Parameters:

U - data type for output() output

All Implemented Interfaces:

Iterable<Operand<U>>, Op

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

Returns shape of tensors.

This operation returns N 1-D integer tensors representing shape of `input[i]s`.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T> ShapeN<Integer>	create(Scope scope, Iterable<Operand<T>> input) Factory method to create a class wrapping a new ShapeN operation using default output types.
static <U extends Number,T> ShapeN<U>	create(Scope scope, Iterable<Operand<T>> input, Class<U> outType) Factory method to create a class wrapping a new ShapeN operation.
Iterator<Operand<U>>	iterator()
List<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

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Method Detail

create

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

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

Parameters:

scope - current scope

input -

outType -

Returns:

a new instance of ShapeN

create

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

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

Parameters:

scope - current scope

input -

Returns:

a new instance of ShapeN

output

public List<Output<U>> output()

iterator

public Iterator<Operand<U>> iterator()

Specified by:

iterator in interface Iterable<Operand<U extends Number>>