Created by: starsblinking
大小为2的全连接层(即size=2的fluid.layers.fc)的返回值应该是个shape为[-1,2]的Tensor吧?(还是lod_tensor?求解答) 我想取出这个2维tensor的第二列作为预测目标放进save_inference_model的target_vars, 请问如何取出这个第二列呢(得到一个shape为[-1,1]的Tensor)?