LearningModelSessionOptions.OverrideNamedDimension(String, UInt32) 메서드

정의

네임스페이스:

Windows.AI.MachineLearning

학습 모델의 명명된 입력 차원을 구체적인 값으로 재정의하는 기능을 제공합니다.

public:
 virtual void OverrideNamedDimension(Platform::String ^ name, unsigned int dimension) = OverrideNamedDimension;

void OverrideNamedDimension(winrt::hstring const& name, uint32_t const& dimension);

public void OverrideNamedDimension(string name, uint dimension);

function overrideNamedDimension(name, dimension)

Public Sub OverrideNamedDimension (name As String, dimension As UInteger)

매개 변수

name

String

Platform::String

winrt::hstring

설정할 입력 차원의 이름입니다.

dimension

UInt32

unsigned int

uint32_t

지정된 차원에 할당할 구체적인 값입니다.

Windows 요구 사항

디바이스 패밀리	Windows 10, version 2104 (10.0.20348.0에서 도입되었습니다.)
API contract	Windows.AI.MachineLearning.MachineLearningContract (v4.0에서 도입되었습니다.)

예제

void SetNamedDimensionOverrides(LearningModel model) {
    // Create LearningModelSessionOptions
    auto options = LearningModelSessionOptions();

    // Override a named input dimension to a concrete value
    options->OverrideNamedDimension(L"dimension_name", 256);

    // Create session
    LearningModelSession session = nullptr;
    session = LearningModelSession(model, LearningModelDeviceKind::GPU, options);
}

설명

이 API를 사용하면 모델 평가 중에 할당되는 세션 생성 중에 텐서의 사전 할당을 허용하므로 성능이 향상될 수 있습니다.