Método ITensorStaticsNative.CreateFromD3D12Resource

Artigo
03/14/2024

Cria um objeto tensor (TensorFloat, TensorInt32Bit) a partir de um ID3D12Resource especificado pelo usuário.

HRESULT CreateFromD3D12Resource(
    ID3D12Resource *value,
    [size_is(shapeCount)] __int64 *shape,
    int shapeCount,
    [out] IUnknown ** result);

Parâmetros

Nome	Tipo	Descrição
value	ID3D12Resource*	O ID3D12Resource a partir do qual criar o tensor.
shape	__int64*	A forma do tensor.
shapeCount	int	O número de dimensões do tensor.
result	IUnknown**	O tensor resultante.

Retornos

HRESULT O resultado da operação.

Exemplos

TensorFloat SoftwareBitmapToDX12Tensor(SoftwareBitmap softwareBitmap)
{
    // ...

    // GPU tensorize
    com_ptr<ITensorStaticsNative> tensorfactory = get_activation_factory<TensorFloat, ITensorStaticsNative>();
    com_ptr<::IUnknown> spUnkTensor;
    TensorFloat input1imagetensor(nullptr);
    int64_t shapes[4] = { 1,3, softwareBitmap.PixelWidth(), softwareBitmap.PixelHeight() };
    CHECK_HRESULT(tensorfactory->CreateFromD3D12Resource(pGPUResource.get(), shapes, 4, spUnkTensor.put()));
    spUnkTensor.try_as(input1imagetensor);

    // ...
}

Confira também

Exemplo de Custom Tensorization

Requisitos

	Requisito
Cliente mínimo com suporte	Windows 10, build 17763
Servidor mínimo com suporte	Windows Server 2019 com Desktop Experience
Cabeçalho	windows.ai.machinelearning.native.h

Observação

Use os recursos a seguir para obter ajuda com o Windows ML:

Para fazer perguntas ou responder a perguntas técnicas sobre o Windows ML, use a marca windows-machine-learning no Stack Overflow.
Para relatar um bug, registre um problema no nosso GitHub.

Compartilhar via