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

DML_SPLIT_OPERATOR_DESC

構造体
サイズx64: 32 バイト / x86: 16 バイト

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

フィールド

フィールドサイズx64x86説明
InputTensorDML_TENSOR_DESC*8/4+0+0複数の出力テンソルに分割する対象のテンソルです。
OutputCountDWORD4+8+4このフィールドは OutputTensors 配列のサイズを決定します。この値は 0 より大きくなければなりません。
OutputTensorsDML_TENSOR_DESC*8/4+16+8入力テンソルから分割された各テンソルの記述を格納する配列です。出力のサイズは、分割軸を除いて入力テンソルと同じサイズでなければなりません。
AxisDWORD4+24+12分割する入力テンソルの次元のインデックスです。すべての入力テンソルと出力テンソルは、この軸を除くすべての次元で同一のサイズでなければなりません。この値は [0, InputTensor.DimensionCount - 1] の範囲内でなければなりません。

公式ドキュメント

入力テンソルを、ある軸に沿って複数の出力テンソルに分割します。

すべての入力テンソルと出力テンソルは、分割軸を除いて同じサイズでなければなりません。分割軸における入力テンソルのサイズによって、可能な分割が決まります。たとえば、入力テンソルの分割軸のサイズが 3 の場合、次のような分割が考えられます。1+1+1 (出力 3 個)、1+2 (出力 2 個)、2+1 (出力 2 個)、または 3 (出力 1 個。これは単に入力テンソルのコピーです)。出力テンソルの分割軸のサイズの合計は、入力テンソルの分割軸のサイズと正確に一致しなければなりません。これらの制約を以下の疑似コードに示します。

splitSize = 0;

for (i = 0; i < OutputCount; i++) {
    assert(outputTensors[i]->DimensionCount == inputTensor->DimensionCount);
    for (dim = 0; dim < inputTensor->DimensionCount; dim++) {
        if (dim == Axis) { splitSize += outputTensors[i]->Sizes[dim]; }
        else { assert(outputTensors[i]->Sizes[dim] == inputTensor->Sizes[dim]); }
    }
}

assert(splitSize == inputTensor->Sizes[Axis]);

単一の出力テンソルに分割した場合は、単に入力テンソルのコピーが生成されます。

この演算子は DML_JOIN_OPERATOR_DESC の逆の演算です。

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

各言語での定義

#include <windows.h>

// DML_SPLIT_OPERATOR_DESC  (x64 32 / x86 16 バイト)
typedef struct DML_SPLIT_OPERATOR_DESC {
    DML_TENSOR_DESC* InputTensor;
    DWORD OutputCount;
    DML_TENSOR_DESC* OutputTensors;
    DWORD Axis;
} DML_SPLIT_OPERATOR_DESC;
using System;
using System.Runtime.InteropServices;

[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
public struct DML_SPLIT_OPERATOR_DESC
{
    public IntPtr InputTensor;
    public uint OutputCount;
    public IntPtr OutputTensors;
    public uint Axis;
}
Imports System.Runtime.InteropServices

<StructLayout(LayoutKind.Sequential, CharSet:=CharSet.Unicode)>
Public Structure DML_SPLIT_OPERATOR_DESC
    Public InputTensor As IntPtr
    Public OutputCount As UInteger
    Public OutputTensors As IntPtr
    Public Axis As UInteger
End Structure
import ctypes
from ctypes import wintypes

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

type DML_SPLIT_OPERATOR_DESC struct {
	InputTensor uintptr
	OutputCount uint32
	OutputTensors uintptr
	Axis uint32
}
type
  DML_SPLIT_OPERATOR_DESC = record
    InputTensor: Pointer;
    OutputCount: DWORD;
    OutputTensors: Pointer;
    Axis: DWORD;
  end;
const DML_SPLIT_OPERATOR_DESC = extern struct {
    InputTensor: ?*anyopaque,
    OutputCount: u32,
    OutputTensors: ?*anyopaque,
    Axis: u32,
};
type
  DML_SPLIT_OPERATOR_DESC {.bycopy.} = object
    InputTensor: pointer
    OutputCount: uint32
    OutputTensors: pointer
    Axis: uint32
struct DML_SPLIT_OPERATOR_DESC
{
    void* InputTensor;
    uint OutputCount;
    void* OutputTensors;
    uint Axis;
}

HSP用 定義

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

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

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