Dela via

Om åtgärden

  • Artikel

Elementwise selecting of one of two inputs given a condition( Elementwise selecting of one of two inputs given a condition).

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

Parametrar

  • condition: villkor enligt vilket elementvärden som väljs
  • thenValue: elementvärde markerat om condition elementet inte är 0
  • elseValue: elementvärde markerat om condition elementet är 0

Glesa värden stöds för närvarande inte.

Returvärde

En tensor av dimensionen för indata. Om någon av indata har dimensionerna 1 tillämpas sändningen. I så fall blir utdatadimensionen det högsta över motsvarande tre argument dimensioner.

Description

If() väljer element från två indata baserat på ett villkor, på ett elementmässigt sätt. För varje indataelement där condition är icke-0 väljs motsvarande element från thenValue , och där condition är 0 väljs motsvarande elseValue element.

Den här funktionen stöder sändning. Det är till exempel möjligt att villkoret är en skalär, eller att en av indata är en konstant tensor utan tidsdimension.

Exempel

Elementmässigt kan maximalt två indata beräknas som en kombination av Greater() och If():

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

Detta fungerar också med sändning. Till exempel kan den linjära likriktaren skrivas med detta med hjälp av en skalär konstant som den andra indatan:

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