U
- data type for output()
output@Operator public final class ShapeN<U extends Number> extends PrimitiveOp implements Iterable<Operand<U>>
This operation returns N 1-D integer tensors representing shape of `input[i]s`.
operation
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> |
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() |
equals, hashCode, op, toString
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
forEach, spliterator
public static <U extends Number,T> ShapeN<U> create(Scope scope, Iterable<Operand<T>> input, Class<U> outType)
scope
- current scopeinput
- outType
- public static <T> ShapeN<Integer> create(Scope scope, Iterable<Operand<T>> input)
scope
- current scopeinput
- Copyright © 2022. All rights reserved.