通过

你当前正在访问 Microsoft Azure Global Edition 技术文档网站。 如果需要访问由世纪互联运营的 Microsoft Azure 中国技术文档网站，请访问 https://docs.azure.cn。

If 操作

  • 项目

在给定条件的情况下，按元素方式选择两个输入之一。

BS.Boolean.If (condition, thenValue, elseValue)

parameters

  • condition：根据所选元素值的条件
  • thenValue：如果 condition 元素不是 0，则选择的元素值
  • elseValue：如果 condition 元素为 0，则选择的元素值

目前不支持稀疏值。

返回值

输入维度的张量。 如果任一输入的维度为 1，则应用广播;在这种情况下，输出维度将成为相应三个参数维度的最大值。

说明

If() 根据条件（按元素方式）从两个输入中选择元素。 对于非 0 的每个输入元素 condition ，都会选择相应的 thenValue 元素;如果为 condition 0，则选择相应的 elseValue 元素。

此函数支持广播。 例如，条件可能是标量，或者其中一个输入是无时间维度的常量张量。

示例

两个输入的元素最大值可以计算为以下两个输入的组合：Greater()If()

MyElementwiseMax (a, b) = BS.Boolean.If (Greater (a, b), a, b)

这也适用于广播。 例如，可以使用标量常量作为第二个输入来编写线性整流器：

MyReLU (x) = MyElementwiseMax (x, Constant(0))