Win32 API 日本語リファレンス
ホーム › AI.MachineLearning.DirectML › DML_ARGMAX_OPERATOR_DESC

DML_ARGMAX_OPERATOR_DESC

構造体
📘 Microsoft Learn (日本語) · English
サイズx64: 40 バイト / x86: 20 バイト

サイズ=各フィールドのバイト数(x64/x86 で異なる場合は x64/x86 と併記)。x64/x86 列=フィールドのバイトオフセット(HSPで dupptr / lpoke / wpoke 等に使用)。

フィールド

フィールド型サイズx64x86説明
InputTensorDML_TENSOR_DESC*8/4+0+0読み取り元のテンソルです。
OutputTensorDML_TENSOR_DESC*8/4+8+4

結果を書き込むテンソルです。各出力要素は、InputTensor の要素の部分集合に対する argmax リダクションの結果です。

  • DimensionCount は InputTensor.DimensionCount と一致していなければなりません (入力テンソルのランクは保持されます)。
  • Sizes は InputTensor.Sizes と一致していなければなりませんが、リダクション対象の Axes に含まれるディメンションはサイズ 1 でなければなりません。
AxisCountDWORD4+16+8リダクションを行う軸の数です。このフィールドは Axes 配列のサイズを決定します。
AxesDWORD*8/4+24+12リダクションを行う軸です。値は [0, InputTensor.DimensionCount - 1] の範囲でなければなりません。
AxisDirectionDML_AXIS_DIRECTION4+32+16

複数の入力要素が同じ値を持つ場合に、どのインデックスを選択するかを決定します。

公式ドキュメント

入力テンソルの 1 つ以上のディメンションにおいて、最大値を持つ要素のインデックスを出力します。

各出力要素は、入力テンソルの要素の部分集合に対して argmax リダクションを適用した結果です。argmax 関数は、入力要素の集合の中で最大値を持つ要素のインデックスを出力します。各リダクションに関与する入力要素は、指定された入力軸によって決まります。同様に、各出力インデックスも指定された入力軸を基準とします。すべての入力軸が指定された場合、この演算子は単一の argmax リダクションを適用し、1 つの出力要素を生成します。

解説(Remarks)

出力テンソルのサイズは入力テンソルのサイズと同じでなければなりませんが、リダクション対象の軸は 1 でなければなりません。

AxisDirection が DML_AXIS_DIRECTION_INCREASING の場合、この API は DML_REDUCE_FUNCTION_ARGMAX を指定した DML_REDUCE_OPERATOR_DESC と同等です。

この機能の一部は DML_REDUCE_OPERATOR_DESC 演算子を通じて公開されており、それ以前の DirectML 機能レベルでもサポートされます。

出典・ライセンス: 上記「公式ドキュメント」の内容は Microsoft の Win32 API ドキュメント(MicrosoftDocs/sdk-api)を日本語に翻訳・改変したものです。© Microsoft Corporation. CC BY 4.0 で提供。
Microsoft 公式リファレンス: 英語 (en-us) · 日本語 (ja-jp) · 原文ソース (GitHub)

各言語での定義

#include <windows.h>

// DML_ARGMAX_OPERATOR_DESC  (x64 40 / x86 20 バイト)
typedef struct DML_ARGMAX_OPERATOR_DESC {
    DML_TENSOR_DESC* InputTensor;
    DML_TENSOR_DESC* OutputTensor;
    DWORD AxisCount;
    DWORD* Axes;
    DML_AXIS_DIRECTION AxisDirection;
} DML_ARGMAX_OPERATOR_DESC;
using System;
using System.Runtime.InteropServices;

[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
public struct DML_ARGMAX_OPERATOR_DESC
{
    public IntPtr InputTensor;
    public IntPtr OutputTensor;
    public uint AxisCount;
    public IntPtr Axes;
    public int AxisDirection;
}
Imports System.Runtime.InteropServices

<StructLayout(LayoutKind.Sequential, CharSet:=CharSet.Unicode)>
Public Structure DML_ARGMAX_OPERATOR_DESC
    Public InputTensor As IntPtr
    Public OutputTensor As IntPtr
    Public AxisCount As UInteger
    Public Axes As IntPtr
    Public AxisDirection As Integer
End Structure
import ctypes
from ctypes import wintypes

class DML_ARGMAX_OPERATOR_DESC(ctypes.Structure):
    _fields_ = [
        ("InputTensor", ctypes.c_void_p),
        ("OutputTensor", ctypes.c_void_p),
        ("AxisCount", wintypes.DWORD),
        ("Axes", ctypes.c_void_p),
        ("AxisDirection", ctypes.c_int),
    ]
#[repr(C)]
pub struct DML_ARGMAX_OPERATOR_DESC {
    pub InputTensor: *mut core::ffi::c_void,
    pub OutputTensor: *mut core::ffi::c_void,
    pub AxisCount: u32,
    pub Axes: *mut core::ffi::c_void,
    pub AxisDirection: i32,
}
import "golang.org/x/sys/windows"

type DML_ARGMAX_OPERATOR_DESC struct {
	InputTensor uintptr
	OutputTensor uintptr
	AxisCount uint32
	Axes uintptr
	AxisDirection int32
}
type
  DML_ARGMAX_OPERATOR_DESC = record
    InputTensor: Pointer;
    OutputTensor: Pointer;
    AxisCount: DWORD;
    Axes: Pointer;
    AxisDirection: Integer;
  end;
const DML_ARGMAX_OPERATOR_DESC = extern struct {
    InputTensor: ?*anyopaque,
    OutputTensor: ?*anyopaque,
    AxisCount: u32,
    Axes: ?*anyopaque,
    AxisDirection: i32,
};
type
  DML_ARGMAX_OPERATOR_DESC {.bycopy.} = object
    InputTensor: pointer
    OutputTensor: pointer
    AxisCount: uint32
    Axes: pointer
    AxisDirection: int32
struct DML_ARGMAX_OPERATOR_DESC
{
    void* InputTensor;
    void* OutputTensor;
    uint AxisCount;
    void* Axes;
    int AxisDirection;
}

HSP用 定義

HSP3.7/3.8 は構造体機能が無いため4byte整数配列(dim)+peek/poke で操作(32/64bitでサイズ・位置が異なる場合はタブで分割)。IronHSP は NSTRUCT(#defstruct/stdim/->)で32/64bit共通。

; HSP3.7/3.8 は構造体機能が無いため、4byte整数の配列変数で操作します。(x86 レイアウト)
; DML_ARGMAX_OPERATOR_DESC サイズ: 20 バイト(x86)
dim st, 5    ; 4byte整数×5(構造体サイズ 20 / 4 切り上げ)
; InputTensor : DML_TENSOR_DESC* (+0, 4byte)  varptr(st)+0 を基点に操作(4byte:入れ子/配列)
; OutputTensor : DML_TENSOR_DESC* (+4, 4byte)  varptr(st)+4 を基点に操作(4byte:入れ子/配列)
; AxisCount : DWORD (+8, 4byte)  st.2 = 値  /  値 = st.2   (lpoke/lpeek も可)
; Axes : DWORD* (+12, 4byte)  st.3 = 値  /  値 = st.3   (lpoke/lpeek も可)
; AxisDirection : DML_AXIS_DIRECTION (+16, 4byte)  st.4 = 値  /  値 = st.4   (lpoke/lpeek も可)
; ※4byte境界の整数は添字 st.N(N=オフセット/4)で読み書き可。それ以外は peek/poke 系を使用。
; HSP3.7/3.8 は構造体機能が無いため、4byte整数の配列変数で操作します。(x64 レイアウト)
; DML_ARGMAX_OPERATOR_DESC サイズ: 40 バイト(x64)
dim st, 10    ; 4byte整数×10(構造体サイズ 40 / 4 切り上げ)
; InputTensor : DML_TENSOR_DESC* (+0, 8byte)  varptr(st)+0 を基点に操作(8byte:入れ子/配列)
; OutputTensor : DML_TENSOR_DESC* (+8, 8byte)  varptr(st)+8 を基点に操作(8byte:入れ子/配列)
; AxisCount : DWORD (+16, 4byte)  st.4 = 値  /  値 = st.4   (lpoke/lpeek も可)
; Axes : DWORD* (+24, 8byte)  qpoke st,24,値 / qpeek(st,24)  ※IronHSPのみ。3.7/3.8は lpoke st,24,下位 : lpoke st,28,上位
; AxisDirection : DML_AXIS_DIRECTION (+32, 4byte)  st.8 = 値  /  値 = st.8   (lpoke/lpeek も可)
; ※4byte境界の整数は添字 st.N(N=オフセット/4)で読み書き可。それ以外は peek/poke 系を使用。
; IronHSP は NSTRUCT(構造体)をサポート。32bit/64bit どちらでも同じコードで動作します。
#defstruct global DML_ARGMAX_OPERATOR_DESC
    #field intptr InputTensor
    #field intptr OutputTensor
    #field int AxisCount
    #field intptr Axes
    #field int AxisDirection
#endstruct

stdim st, DML_ARGMAX_OPERATOR_DESC        ; NSTRUCT 変数を確保
st->AxisCount = 100
mes "AxisCount=" + st->AxisCount