當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.raw_ops.StatelessCase用法及代碼示例

調用單個分支函數的n-way switch 語句。

用法

tf.raw_ops.StatelessCase(
    branch_index, input, Tout, branches, output_shapes=[], name=None
)

參數

branch_index Tensor 類型為 int32 。分支選擇器，一個 int32 張量。
input Tensor 對象的列表。傳遞給分支函數的輸入張量列表。
Tout tf.DTypes 的列表。輸出類型列表。
branches 用 @Defun 修飾的函數列表，長度為 >= 1 。一個函數列表，每個函數都接受 'inputs' 並返回一個張量列表，其類型與每個其他分支返回的類型相同。
output_shapes 可選的形狀列表(每個 tf.TensorShape 或 ints 列表)。默認為 [] 。
name 操作的名稱(可選)。

返回

類型為 Tout 的 Tensor 對象的列表。

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.

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.raw_ops.StatelessCase。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。