struktur DML_ACTIVATION_SHRINK_OPERATOR_DESC (directml.h)

  • Artikel

Melakukan fungsi aktivasi penyusutan pada setiap elemen di InputTensor, menempatkan hasilnya ke elemen OutputTensor yang sesuai.

f(x) = x - Bias, if x > Threshold
       x + Bias, if x < -Threshold
       0,        otherwise

Operator ini mendukung eksekusi di tempat, yang berarti bahwa tensor output diizinkan untuk alias InputTensor selama pengikatan.

Sintaks

struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Bias;
  FLOAT                 Threshold;
};

Anggota

InputTensor

Jenis: const DML_TENSOR_DESC*

Tensor input untuk dibaca.

OutputTensor

Jenis: const DML_TENSOR_DESC*

Tensor output untuk menulis hasilnya.

Bias

Jenis: FLOAT

Nilai bias. Default umum untuk nilai ini adalah 0,0.

Threshold

Jenis: FLOAT

Nilai ambang batas. Default umum untuk nilai ini adalah 0,5.

Keterangan

Ketersediaan

Operator ini diperkenalkan di DML_FEATURE_LEVEL_2_0.

Batasan tensor

InputTensor dan OutputTensor harus memiliki DataType, DimensionCount, dan Sizes yang sama.

Dukungan tensor

DML_FEATURE_LEVEL_5_1 ke atas

Tensor Jenis Jumlah dimensi yang didukung Tipe data yang didukung
InputTensor Input 1 hingga 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor Output 1 hingga 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 ke atas

Tensor Jenis Jumlah dimensi yang didukung Tipe data yang didukung
InputTensor Input 1 hingga 8 FLOAT32, FLOAT16
OutputTensor Output 1 hingga 8 FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_0 ke atas

Tensor Jenis Jumlah dimensi yang didukung Tipe data yang didukung
InputTensor Input 4 hingga 5 FLOAT32, FLOAT16
OutputTensor Output 4 hingga 5 FLOAT32, FLOAT16

Persyaratan

Persyaratan Nilai
Klien minimum yang didukung Windows 10, versi 2004 (10.0; Build 19041)
Server minimum yang didukung Windows Server, versi 2004 (10.0; Build 19041)
Header directml.h