如何使用张量流在完全连接的层中还原

Question

train.py：

prediction = tf.contrib.layers.fully_connected(outputs[:, -1, :], 
                                output_dim, activation_fn=None, scope="prediction")

test.py：

prediction = graph.get_tensor_by_name("prediction:0")

从train.py保存图形后，我想在 test.py 中恢复预测变量，但不起作用。

' tf.contrib.layers.fully_connected '没有 name 参数。只有 scope 参数。我该如何恢复？

Answer 1

获取tensorflow为您定义的全连接层的名称：

train.py

prediction/BiasAdd:0

类似于- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath { if(self.tableView == tableView){ NSString *identifier = [self identifierForCellAtIndexPath:indexPath]; UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:identifier forIndexPath:indexPath]; cell.selectionStyle = UITableViewCellSelectionStyleNone; cell.textLabel.text = @"The text you want to display"; return cell; } return nil; }

在test.py中使用该名称正确还原变量