ホーム › AI.MachineLearning.DirectML › DML_GATHER_OPERATOR_DESC
DML_GATHER_OPERATOR_DESC
構造体サイズ=各フィールドのバイト数(x64/x86 で異なる場合は x64/x86 と併記)。x64/x86 列=フィールドのバイトオフセット(HSPで dupptr / lpoke / wpoke 等に使用)。
フィールド
| フィールド | 型 | サイズ | x64 | x86 | 説明 |
|---|---|---|---|---|---|
| InputTensor | DML_TENSOR_DESC* | 8/4 | +0 | +0 | 読み取り元のテンソルです。 |
| IndicesTensor | DML_TENSOR_DESC* | 8/4 | +8 | +4 | インデックスを格納するテンソルです。このテンソルの DimensionCount は InputTensor.DimensionCount と一致していなければなりません。
無効なインデックスを指定すると出力は不正になりますが、失敗は発生せず、すべての読み取りは入力テンソルのメモリ内に安全にクランプされます。 |
| OutputTensor | DML_TENSOR_DESC* | 8/4 | +16 | +8 | 結果を書き込むテンソルです。このテンソルの DimensionCount と DataType は InputTensor.DimensionCount と一致していなければなりません。想定される OutputTensor.Sizes は、現在の Axis で分割した InputTensor.Sizes の先頭側セグメントと末尾側セグメントの間に、IndicesTensor.Sizes を挿入して連結したものです。 各次元は右詰めで整列され、そのままでは出力の DimensionCount を超過してしまう入力サイズの先頭の 1 は切り捨てられます。 このテンソルで意味を持つ次元の数は、IndexDimensions と InputTensor の 元のランク によって決まります。元のランクとは、先頭に 1 を埋める前の次元数です。出力で意味を持つ次元の数は、InputTensor の 元のランク + IndexDimensions - 1 で計算できます。この値は OutputTensor の DimensionCount 以下でなければなりません。 |
| Axis | DWORD | 4 | +24 | +12 | 収集の対象とする InputTensor の軸の次元です。範囲は [0, *InputTensor.DimensionCount*) です。 |
| IndexDimensions | DWORD | 4 | +28 | +16 | IndicesTensor 内で、無関係な先頭の 1 を無視した後の実際のインデックス次元の数です。範囲は [0, IndicesTensor.DimensionCount) です。たとえば、IndicesTensor.Sizes = { 1, 1, 4, 6 } で IndexDimensions = 3 の場合、実際に意味を持つインデックスは { 1, 4, 6 } です。 |
公式ドキュメント
Axis に沿って入力テンソルから要素を収集し、IndicesTensor を使用してインデックスを再マップします。この演算子は次の擬似コードの処理を実行します。ここで "..." は一連の座標を表し、正確な動作は軸とインデックスの次元数によって決まります。
output[...] = input[..., indices[...], ...]
出典・ライセンス: 上記「公式ドキュメント」の内容は Microsoft の Win32 API ドキュメント(MicrosoftDocs/sdk-api)を日本語に翻訳・改変したものです。© Microsoft Corporation. CC BY 4.0 で提供。
Microsoft 公式リファレンス: 英語 (en-us) · 日本語 (ja-jp) · 原文ソース (GitHub)
Microsoft 公式リファレンス: 英語 (en-us) · 日本語 (ja-jp) · 原文ソース (GitHub)
各言語での定義
#include <windows.h>
// DML_GATHER_OPERATOR_DESC (x64 32 / x86 20 バイト)
typedef struct DML_GATHER_OPERATOR_DESC {
DML_TENSOR_DESC* InputTensor;
DML_TENSOR_DESC* IndicesTensor;
DML_TENSOR_DESC* OutputTensor;
DWORD Axis;
DWORD IndexDimensions;
} DML_GATHER_OPERATOR_DESC;using System;
using System.Runtime.InteropServices;
[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
public struct DML_GATHER_OPERATOR_DESC
{
public IntPtr InputTensor;
public IntPtr IndicesTensor;
public IntPtr OutputTensor;
public uint Axis;
public uint IndexDimensions;
}Imports System.Runtime.InteropServices
<StructLayout(LayoutKind.Sequential, CharSet:=CharSet.Unicode)>
Public Structure DML_GATHER_OPERATOR_DESC
Public InputTensor As IntPtr
Public IndicesTensor As IntPtr
Public OutputTensor As IntPtr
Public Axis As UInteger
Public IndexDimensions As UInteger
End Structureimport ctypes
from ctypes import wintypes
class DML_GATHER_OPERATOR_DESC(ctypes.Structure):
_fields_ = [
("InputTensor", ctypes.c_void_p),
("IndicesTensor", ctypes.c_void_p),
("OutputTensor", ctypes.c_void_p),
("Axis", wintypes.DWORD),
("IndexDimensions", wintypes.DWORD),
]#[repr(C)]
pub struct DML_GATHER_OPERATOR_DESC {
pub InputTensor: *mut core::ffi::c_void,
pub IndicesTensor: *mut core::ffi::c_void,
pub OutputTensor: *mut core::ffi::c_void,
pub Axis: u32,
pub IndexDimensions: u32,
}import "golang.org/x/sys/windows"
type DML_GATHER_OPERATOR_DESC struct {
InputTensor uintptr
IndicesTensor uintptr
OutputTensor uintptr
Axis uint32
IndexDimensions uint32
}type
DML_GATHER_OPERATOR_DESC = record
InputTensor: Pointer;
IndicesTensor: Pointer;
OutputTensor: Pointer;
Axis: DWORD;
IndexDimensions: DWORD;
end;const DML_GATHER_OPERATOR_DESC = extern struct {
InputTensor: ?*anyopaque,
IndicesTensor: ?*anyopaque,
OutputTensor: ?*anyopaque,
Axis: u32,
IndexDimensions: u32,
};type
DML_GATHER_OPERATOR_DESC {.bycopy.} = object
InputTensor: pointer
IndicesTensor: pointer
OutputTensor: pointer
Axis: uint32
IndexDimensions: uint32struct DML_GATHER_OPERATOR_DESC
{
void* InputTensor;
void* IndicesTensor;
void* OutputTensor;
uint Axis;
uint IndexDimensions;
}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_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:入れ子/配列)
; Axis : DWORD (+12, 4byte) st.3 = 値 / 値 = st.3 (lpoke/lpeek も可)
; IndexDimensions : DWORD (+16, 4byte) st.4 = 値 / 値 = st.4 (lpoke/lpeek も可)
; ※4byte境界の整数は添字 st.N(N=オフセット/4)で読み書き可。それ以外は peek/poke 系を使用。; HSP3.7/3.8 は構造体機能が無いため、4byte整数の配列変数で操作します。(x64 レイアウト)
; DML_GATHER_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:入れ子/配列)
; Axis : DWORD (+24, 4byte) st.6 = 値 / 値 = st.6 (lpoke/lpeek も可)
; IndexDimensions : DWORD (+28, 4byte) st.7 = 値 / 値 = st.7 (lpoke/lpeek も可)
; ※4byte境界の整数は添字 st.N(N=オフセット/4)で読み書き可。それ以外は peek/poke 系を使用。; IronHSP は NSTRUCT(構造体)をサポート。32bit/64bit どちらでも同じコードで動作します。
#defstruct global DML_GATHER_OPERATOR_DESC
#field intptr InputTensor
#field intptr IndicesTensor
#field intptr OutputTensor
#field int Axis
#field int IndexDimensions
#endstruct
stdim st, DML_GATHER_OPERATOR_DESC ; NSTRUCT 変数を確保
st->Axis = 100
mes "Axis=" + st->Axis