我有一段代码返回一个形状为[1,2]的[] [] float32值。以下是代码段。
result, runErr := model.Session.Run(
map[tf.Output]*tf.Tensor{
model.Graph.Operation("x").Output(0): tensor,
},
[]tf.Output{
model.Graph.Operation("predict").Output(0),
},
nil,
)
fmt.Println(result[0].Value())
以下代码的输出是:
[[0 1]]
我需要访问单个元素0和1,但是当我尝试使用以下fmt.Println(result[0].Value()[0][0])
获取它时会导致错误:
(type interface {}不支持索引)
如何访问单个元素以进行处理?
答案 0 :(得分:1)
使用类型断言,我们可以通过索引
获取单个值 md := result[0].Value().([][]float32)
fmt.Println(md[0][0])
fmt.Println(md[0][1])