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

DML_GATHER_ND1_OPERATOR_DESC

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

サイズ=各フィールドのバイト数(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 の最後の次元は、実際にはインデックスタプル 1 つあたりの座標数であり、InputTensor.DimensionCount を超えることはできません。たとえば、Sizes が {1,4,5,2} で IndicesDimensionCount = 3 のインデックステンソルは、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 // 3 座標タプルの {1,2} 配列と考えることができます

// {1,2} はインデックステンソル (タプルのサイズを表す最後の次元を除く) に由来し、
// {6,7} は入力テンソルの先頭 3 次元を除いた部分に由来します。
// これは、インデックスタプルが 3 要素 (インデックステンソルの最後の次元) であるためです。
OutputTensor.Sizes = {1, 1,2,6,7}
InputDimensionCountDWORD4+24+12InputTensor 内で、無関係な先頭の次元を無視した後の実際の入力次元数であり、範囲は [1, *InputTensor.DimensionCount*] です。たとえば、InputTensor.Sizes = {1,1,4,6} かつ InputDimensionCount = 3 の場合、実際に意味を持つインデックスは {1,4,6} です。
IndicesDimensionCountDWORD4+28+16IndicesTensor 内で、無関係な先頭の次元を無視した後の実際のインデックス次元数であり、範囲は [1, IndicesTensor.DimensionCount] です。たとえば、IndicesTensor.Sizes = {1,1,4,6} かつ IndicesDimensionCount = 3 の場合、実際に意味を持つインデックスは {1,4,6} です。
BatchDimensionCountDWORD4+32+20各テンソル (InputTensor、IndicesTensor、OutputTensor) のうち、独立したバッチとみなされる次元数であり、[0, InputTensor.DimensionCount) と [0, IndicesTensor.DimensionCount) の両方の範囲に収まります。バッチ数は 0 でもよく、その場合は単一のバッチを意味します。たとえば、IndicesTensor.Sizes = {1,3,4,5,6,7}、IndicesDimensionCount = 5、BatchDimensionCount = 2 の場合、バッチは {3,4}、意味を持つインデックスは {5,6,7} となります。

公式ドキュメント

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

output[batch, ...] = input[batch, indices[batch, ...], ...]

解説(Remarks)

DML_GATHER_ND1_OPERATOR_DESC は BatchDimensionCount を追加したものであり、BatchDimensionCount = 0 の場合は DML_GATHER_ND_OPERATOR_DESC と同等です。

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

各言語での定義

#include <windows.h>

// DML_GATHER_ND1_OPERATOR_DESC  (x64 40 / x86 24 バイト)
typedef struct DML_GATHER_ND1_OPERATOR_DESC {
    DML_TENSOR_DESC* InputTensor;
    DML_TENSOR_DESC* IndicesTensor;
    DML_TENSOR_DESC* OutputTensor;
    DWORD InputDimensionCount;
    DWORD IndicesDimensionCount;
    DWORD BatchDimensionCount;
} DML_GATHER_ND1_OPERATOR_DESC;
using System;
using System.Runtime.InteropServices;

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

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

class DML_GATHER_ND1_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),
        ("BatchDimensionCount", wintypes.DWORD),
    ]
#[repr(C)]
pub struct DML_GATHER_ND1_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,
    pub BatchDimensionCount: u32,
}
import "golang.org/x/sys/windows"

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

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_ND1_OPERATOR_DESC サイズ: 24 バイト(x86)
dim st, 6    ; 4byte整数×6(構造体サイズ 24 / 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 も可)
; BatchDimensionCount : DWORD (+20, 4byte)  st.5 = 値  /  値 = st.5   (lpoke/lpeek も可)
; ※4byte境界の整数は添字 st.N(N=オフセット/4)で読み書き可。それ以外は peek/poke 系を使用。
; HSP3.7/3.8 は構造体機能が無いため、4byte整数の配列変数で操作します。(x64 レイアウト)
; DML_GATHER_ND1_OPERATOR_DESC サイズ: 40 バイト(x64)
dim st, 10    ; 4byte整数×10(構造体サイズ 40 / 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 も可)
; BatchDimensionCount : DWORD (+32, 4byte)  st.8 = 値  /  値 = st.8   (lpoke/lpeek も可)
; ※4byte境界の整数は添字 st.N(N=オフセット/4)で読み書き可。それ以外は peek/poke 系を使用。
; IronHSP は NSTRUCT(構造体)をサポート。32bit/64bit どちらでも同じコードで動作します。
#defstruct global DML_GATHER_ND1_OPERATOR_DESC
    #field intptr InputTensor
    #field intptr IndicesTensor
    #field intptr OutputTensor
    #field int InputDimensionCount
    #field int IndicesDimensionCount
    #field int BatchDimensionCount
#endstruct

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