ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

ShapeN

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>
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>>
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

public Iterator<Operand<U>> iterator ()

public List<Output<U>> output ()