LearningModelSessionOptions.OverrideNamedDimension(String, UInt32) 方法
定义
重要
一些信息与预发行产品相关,相应产品在发行之前可能会进行重大修改。 对于此处提供的信息,Microsoft 不作任何明示或暗示的担保。
提供将学习模型的命名输入维度重写为具体值的功能。
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 可提高性能,因为它允许在会话创建期间预先分配张量,否则会在模型评估期间分配这些张量。