Spec-Zone.ru › TensorFlow

tf.raw_ops.StatelessCase

Оператор переключения с n вариантами, вызывающий одну функцию ветви.

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.StatelessCase

tf.raw_ops.StatelessCase(
    branch_index, input, Tout, branches, output_shapes=[], name=None
)
An n-way switch statement, implementing the following:

```
switch (branch_index) {
  case 0:
    output = branches[0](input);
    break;
  case 1:
    output = branches[1](input);
    break;
  ...
  case [[nbranches-1]]:
  default:
    output = branches[nbranches-1](input);
    break;
}
```

This should only be used when the none of branches has stateful ops.
Аргументы
branch_index Tensor типа int32. Селектор ветви, тензор int32.
input Список объектов Tensor. Список входных тензоров, передаваемых функции ветви.
Tout Список tf.DTypes. Список выходных типов.
branches Список функций, декорированных с помощью @Defun; его длина равна >= 1. Каждая функция в списке принимает «inputs» и возвращает список тензоров, типы которых совпадают с типами тензоров, возвращаемых всеми остальными ветвями.
output_shapes Необязательный список форм (каждый элемент — tf.TensorShape или список ints). По умолчанию — [].
name Имя операции (необязательно).
Возвращает
Список объектов Tensor типа Tout.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/StatelessCase

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API