DML_ELEMENT_WISE_IF_OPERATOR_DESC-Struktur (directml.h)

Wählt Elemente entweder aus ATensor oder BTensor aus, abhängig vom Wert des entsprechenden Elements in ConditionTensor. Ungleich null Elemente von ConditionTensor wählen aus ATensor aus, während nullwertige Elemente aus BTensor auswählen.

f(cond, a, b) = a, if cond != 0
                b, otherwise

Example:
    [[1, 0], [1, 1]] // ConditionTensor
    [[1, 2], [3, 4]] // ATensor
    [[9, 8], [7, 6]] // BTensor

    [[1, 8], [3, 4]] // Output

Syntax

struct DML_ELEMENT_WISE_IF_OPERATOR_DESC {
  const DML_TENSOR_DESC *ConditionTensor;
  const DML_TENSOR_DESC *ATensor;
  const DML_TENSOR_DESC *BTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Member

ConditionTensor

Typ: const DML_TENSOR_DESC*

Der Bedingungs tensor, aus dem gelesen werden soll.

ATensor

Typ: const DML_TENSOR_DESC*

Ein Tensor, der die linken Eingaben enthält.

BTensor

Typ: const DML_TENSOR_DESC*

Ein Tensor, der die rechten Eingaben enthält.

OutputTensor

Typ: const DML_TENSOR_DESC*

Der Ausgabe tensor, in den die Ergebnisse geschrieben werden sollen.

Hinweise

Kann verwendet werden, um andere Aggregatoperatoren wie LeakyRelu funktionell aufzubauen. Dies ist eine Abbildung im Pseudocode (nicht die effizienteste, aber möglich): LeakyRelu(x) = If(Less(x, 0), Mul(x, alpha), x).

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_2_0eingeführt.

Tensoreinschränkungen

• ATensor, BTensor, ConditionTensor und OutputTensor müssen über die gleichen DimensionCount und Sizes verfügen.
• ATensor, BTensor und OutputTensor müssen denselben DataType aufweisen.

Tensorunterstützung

DML_FEATURE_LEVEL_5_0 und höher

DML_FEATURE_LEVEL_3_0 und höher

DML_FEATURE_LEVEL_2_0 und höher

Anforderungen