如何使用javascript制作TensorProto,以及如何将其更改为dtype?

时间:2018-10-17 13:28:26

标签: javascript tensorflow request

我必须使用tf_serving。我在打字稿和python上找到了示例。打字稿代码有效,很容易将其更改为javascript。

private buildPredictRequest(buffer: Array<any>): Object {
    const request = {
        model_spec: {
            name: this.modelName,
            signature_name: this.signatureName,
        },
        inputs: {
            inputs: {
                dtype: 'DT_STRING',
                tensor_shape: {
                    dim: {
                        size: buffer.length,
                    },
                },
                string_val: buffer,
            },
        },
    };
    return request;
}

这是发送它的功能。它发送一个DT_STRING。但是我需要的是DT_FLOAT,但是名称的简单转换不起作用。

TensorProto如何将其更改为DT_FLOAT?

0 个答案:

没有答案