public final class
ShapeN
Returns shape of tensors.
This operation returns N 1-D integer tensors representing shape of `input[i]s`.
Public Methods
static <T> ShapeN<Integer> | |
static <U extends Number, T> ShapeN<U> | |
Iterator<Operand<U>> |
iterator()
|
List<Output<U>> |
output()
|
Inherited Methods
Public Methods
public static 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 |
---|
Returns
- a new instance of ShapeN
public static 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 |
---|
Returns
- a new instance of ShapeN