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

DML_GATHER_ND_OPERATOR_DESC

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

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

フィールド

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

インデックスを格納するテンソルです。このテンソルの DimensionCount は InputTensor.DimensionCount と一致していなければなりません。IndicesTensor の最後の次元は実際にはインデックスタプルあたりの座標数であり、InputTensor.DimensionCount を超えることはできません。たとえば、Sizes が {1,4,5,2} で IndicesDimensionCount = 3 の indices テンソルは、InputTensor に対してインデックス付けを行う 2 座標タプルの 4x5 配列を意味します。

DML_FEATURE_LEVEL_3_0 以降では、このテンソルに符号付き整数型を使用する場合、この演算子は負のインデックス値をサポートします。負のインデックスは、対応する次元の末尾からの相対位置として解釈されます。たとえば、インデックス -1 はその次元の最後の要素を指します。

OutputTensorDML_TENSOR_DESC*8/4+16+8

結果の書き込み先となるテンソルです。このテンソルの DimensionCount および DataType は InputTensor.DimensionCount と一致していなければなりません。想定される OutputTensor.Sizes は、IndicesTensor.Sizes の先頭部分と InputTensor.Sizes の末尾部分を連結したもので、次のようになります。

indexTupleSize = IndicesTensor.Sizes[IndicesTensor.DimensionCount - 1]
OutputTensor.Sizes = {
    1...,
    IndicesTensor.Sizes[(IndicesTensor.DimensionCount - IndicesDimensionCount) .. (IndicesTensor.DimensionCount - 1)],
    InputTensor.Sizes[(InputTensor.DimensionCount - indexTupleSize) .. InputTensor.DimensionCount]
}

出力の次元は右詰めで配置され、OutputTensor.DimensionCount を満たすために必要に応じて先頭に 1 が補われます。

次に例を示します。

InputTensor.Sizes = {3,4,5,6,7}
InputDimensionCount = 5
IndicesTensor.Sizes = {1,1, 1,2,3}
IndicesDimensionCount = 3 // can be thought of as a {1,2} array of 3-coordinate tuples

// The {1,2} comes from the indices tensor (ignoring last dimension which is the tuple size),
// and the {6,7} comes from input tensor, ignoring the first 3 dimensions
// since the index tuples are 3 elements (from the indices tensor last dimension).
OutputTensor.Sizes = {1, 1,2,6,7}
InputDimensionCountDWORD4+24+12無関係な先頭の次元を無視した後の、InputTensor 内の実際の入力次元数です。範囲は [1, *InputTensor.DimensionCount*] です。たとえば、InputTensor.Sizes = {1,1,4,6} かつ InputDimensionCount = 3 の場合、実際に意味を持つインデックスは {1,4,6} です。
IndicesDimensionCountDWORD4+28+16無関係な先頭の次元を無視した後の、IndicesTensor 内の実際のインデックス次元数です。範囲は [1, IndicesTensor.DimensionCount] です。たとえば、IndicesTensor.Sizes = {1,1,4,6} かつ IndicesDimensionCount = 3 の場合、実際に意味を持つインデックスは {1,4,6} です。

公式ドキュメント

indices テンソルを使用してインデックスを入力のサブブロック全体に再マップしながら、入力テンソルから要素を収集します。この演算子は次の擬似コードを実行します。ここで "..." は一連の座標を表し、正確な動作は入力および indices の次元数によって異なります。

output[...] = input[indices[...]]

解説(Remarks)

この演算子の新しいバージョンである DML_OPERATOR_GATHER_ND1 が DML_FEATURE_LEVEL_3_0 で導入されました。

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

各言語での定義

#include <windows.h>

// DML_GATHER_ND_OPERATOR_DESC  (x64 32 / x86 20 バイト)
typedef struct DML_GATHER_ND_OPERATOR_DESC {
    DML_TENSOR_DESC* InputTensor;
    DML_TENSOR_DESC* IndicesTensor;
    DML_TENSOR_DESC* OutputTensor;
    DWORD InputDimensionCount;
    DWORD IndicesDimensionCount;
} DML_GATHER_ND_OPERATOR_DESC;
using System;
using System.Runtime.InteropServices;

[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
public struct DML_GATHER_ND_OPERATOR_DESC
{
    public IntPtr InputTensor;
    public IntPtr IndicesTensor;
    public IntPtr OutputTensor;
    public uint InputDimensionCount;
    public uint IndicesDimensionCount;
}
Imports System.Runtime.InteropServices

<StructLayout(LayoutKind.Sequential, CharSet:=CharSet.Unicode)>
Public Structure DML_GATHER_ND_OPERATOR_DESC
    Public InputTensor As IntPtr
    Public IndicesTensor As IntPtr
    Public OutputTensor As IntPtr
    Public InputDimensionCount As UInteger
    Public IndicesDimensionCount As UInteger
End Structure
import ctypes
from ctypes import wintypes

class DML_GATHER_ND_OPERATOR_DESC(ctypes.Structure):
    _fields_ = [
        ("InputTensor", ctypes.c_void_p),
        ("IndicesTensor", ctypes.c_void_p),
        ("OutputTensor", ctypes.c_void_p),
        ("InputDimensionCount", wintypes.DWORD),
        ("IndicesDimensionCount", wintypes.DWORD),
    ]
#[repr(C)]
pub struct DML_GATHER_ND_OPERATOR_DESC {
    pub InputTensor: *mut core::ffi::c_void,
    pub IndicesTensor: *mut core::ffi::c_void,
    pub OutputTensor: *mut core::ffi::c_void,
    pub InputDimensionCount: u32,
    pub IndicesDimensionCount: u32,
}
import "golang.org/x/sys/windows"

type DML_GATHER_ND_OPERATOR_DESC struct {
	InputTensor uintptr
	IndicesTensor uintptr
	OutputTensor uintptr
	InputDimensionCount uint32
	IndicesDimensionCount uint32
}
type
  DML_GATHER_ND_OPERATOR_DESC = record
    InputTensor: Pointer;
    IndicesTensor: Pointer;
    OutputTensor: Pointer;
    InputDimensionCount: DWORD;
    IndicesDimensionCount: DWORD;
  end;
const DML_GATHER_ND_OPERATOR_DESC = extern struct {
    InputTensor: ?*anyopaque,
    IndicesTensor: ?*anyopaque,
    OutputTensor: ?*anyopaque,
    InputDimensionCount: u32,
    IndicesDimensionCount: u32,
};
type
  DML_GATHER_ND_OPERATOR_DESC {.bycopy.} = object
    InputTensor: pointer
    IndicesTensor: pointer
    OutputTensor: pointer
    InputDimensionCount: uint32
    IndicesDimensionCount: uint32
struct DML_GATHER_ND_OPERATOR_DESC
{
    void* InputTensor;
    void* IndicesTensor;
    void* OutputTensor;
    uint InputDimensionCount;
    uint IndicesDimensionCount;
}

HSP用 定義

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

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

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