help wanted
仓库指标
- Star
- (2,492 star)
- PR 合并指标
- (PR 指标待抓取)
描述
Hi,
Is there a way to automatically create the mapping for the Placeholder (e.g. Placeholder_1) to the input name (e.g. Height) from the model?
Also is there a way to automatically read the "prob_out" from the model?
[]tf.Output and map[tf.Output]*tf.Tensor are both used for model.Exec:
height, _ := tf.NewTensor([1]int64{"Height"})
width, _ := tf.NewTensor([1]int64{"Width"})
results := model.Exec([]tf.Output{
model.Op("prob_out", 0),
}, map[tf.Output]*tf.Tensor{
model.Op("Placeholder_1", 0): height,
model.Op("Placeholder_2", 0): width,
})
Thanks.