如何从Java Map创建TensorProto

<dependency>
    <groupId>org.tensorflow</groupId>
    <artifactId>tensorflow</artifactId>
    <version>2.7.0</version>
</dependency>

TensorProto.Builder tensorBuilder = TensorProto.newBuilder();

tensorBuilder.setDtype(DataType.DT_FLOAT);

Map<String, Float> dataMap = new HashMap<>();
dataMap.put("key1", 1.0f);
dataMap.put("key2", 2.0f);

for (Map.Entry<String, Float> entry : dataMap.entrySet()) {
    String key = entry.getKey();
    Float value = entry.getValue();

    // 创建一个TensorProto.TensorShapeProto.Dim对象，用于设置Tensor的形状
    TensorShapeProto.Dim dim = TensorShapeProto.Dim.newBuilder().setSize(1).build();

    // 创建一个TensorProto.TensorShapeProto对象，用于设置Tensor的形状
    TensorShapeProto shape = TensorShapeProto.newBuilder().addDim(dim).build();

    // 创建一个TensorProto.TensorProto.TensorContent对象，用于设置Tensor的数据内容
    TensorProto.TensorContent tensorContent = TensorProto.TensorContent.newBuilder()
            .addFloatVal(value)
            .build();

    // 创建一个TensorProto.TensorProto对象，用于设置Tensor的名称、形状和数据内容
    TensorProto tensor = TensorProto.newBuilder()
            .setDtype(DataType.DT_FLOAT)
            .setTensorShape(shape)
            .setTensorContent(tensorContent.toByteString())
            .build();

    // 将TensorProto对象添加到TensorProto.Builder中
    tensorBuilder.putTensorContent(key, tensor);
}

TensorProto tensorProto = tensorBuilder.build();