InfeedEnqueueTuple

publiczna klasa końcowa InfeedEnqueueTuple

Podaje do obliczeń wiele wartości Tensora w postaci krotki XLA.

Klasy zagnieżdżone

klasa InfeedEnqueueTuple.Options Opcjonalne atrybuty dla InfeedEnqueueTuple

Metody publiczne

statyczny InfeedEnqueueTuple
utwórz (zakres zakresu , wejścia Iterable< Operand <?>>, Lista< Kształt > kształty, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację InfeedEnqueueTuple.
statyczne InfeedEnqueueTuple.Options
urządzenieOrdinal (Długie urządzenieOrdinal)
statyczne InfeedEnqueueTuple.Options
układy (układy List<Long>)

Metody dziedziczone

Metody publiczne

public static InfeedEnqueueTuple create ( Zakres zakresu , Iterowalne < Operandy <?>> wejścia, Lista < Kształt > kształty, Opcje... opcje)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację InfeedEnqueueTuple.

Parametry
zakres aktualny zakres
wejścia Lista tensorów, które zostaną dostarczone przy użyciu mechanizmu podawania.
kształty Kształty każdego tensora na wejściach.
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowa instancja InfeedEnqueueTuple

public static InfeedEnqueueTuple.Options urządzenieOrdinal (Długie urządzenieOrdinal)

Parametry
urządzeniePorządek Urządzenie TPU, którego chcesz użyć. Powinno to wynosić -1, gdy Op jest uruchomione na urządzeniu TPU i >= 0, gdy Op działa na urządzeniu z procesorem.

publiczne statyczne układy InfeedEnqueueTuple.Options (układy List<Long>)

Parametry
układy Wektor przechowujący żądany układ w kolejności od mniejszej do większej dla wszystkich kształtów krotki, w kolejności, w jakiej kształty pojawiają się na wejściu „kształtów”. Elementy układu dla podkształtu można ustawić na -1, w takim przypadku odpowiedni układ zostanie obliczony w wyniku operacji dosuwu.