张量流

Question

我试图在张量流图中使用条件随机场损失。

我正在执行序列标记任务：

我有一系列元素作为输入[a，B，C，D]。每个元素可以属于三个不同类中的一个。类以一种热编码方式表示：属于类0的元素由向量[1,0,0]表示。

我的输入标签（y）有大小（batch_sizexsequence_lengthxnum_classes）。

我的网络产生相同形状的日志。

假设我所有的序列都有长度4。

这是我的代码：

import tensorflow as tf

sequence_length = 4
num_classes = 3
input_y = tf.placeholder(tf.int32, shape=[None, sequence_length, num_classes])
logits = tf.placeholder(tf.float32, shape=[None, None, num_classes])
dense_y = tf.argmax(input_y, -1, output_type=tf.int32)

log_likelihood, _ = tf.contrib.crf.crf_log_likelihood(logits, dense_y, sequence_length)

我得到以下错误：

文件“/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/crf/python/ops/crf.py”中的第1行“/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/crf/python/ops/crf.py”，第182行，在crf\u-sequence\u-score-false\fn=\u-multi-seq\fn）文件中“/usr/local/lib/python2。7/dist包/tensorflow/python/layers/utils。py“，第206行，在smart\u cond pred中，true\u fn=true\u fn，false\u fn=false\u fn，name=name）文件“/usr/local/lib/python2。7/dist包/tensorflow/python/framework/smart\u cond。py”，第59行，在smart_cond name=name）文件“/usr/local/lib/python2中。7/dist包/tensorflow/python/util/deprecation。py”，第432行，在新函数return func（*args，**kwargs）文件“/usr/local/lib/python2中。7/dist包/tensorflow/python/ops/control\u flow\u ops。py”，第2063行，在cond orig_res_t中，res_t=context_t.BuildCondBranch（true_fn）文件“/usr/local/lib/python2。7/dist包/tensorflow/python/ops/control\u flow\u ops。py”，第1913行，在BuildCondBranch原始_result=fn（）文件“/usr/local/lib/python2中。7/dist包/tensorflow/contrib/crf/python/ops/crf。py“，第95行，在文件“/usr/local/lib/python2”中的单个fn数组操作concat（[示例索引，标记索引]，axis=1）。7/dist包/tensorflow/python/ops/gen_数组_ops。py“，第2975行，在gather_nd“GatherNd”中，params=params，index=index，name=name）文件“/usr/local/lib/python2。7/dist包/tensorflow/python/framework/op_def_库。py“，第787行，在_apply_op_helper op_def=op_def）文件“/usr/local/lib/python2中。7/dist包/tensorflow/python/framework/ops。py”，第3392行，在create_op_def=op_def）文件“/usr/local/lib/python2中。7/dist包/tensorflow/python/framework/ops。py”，第1734行，在init控制输入（ops）文件“/usr/local/lib/python2中。7/dist包/tensorflow/python/framework/ops。py“，第1570行，在创建操作提升值错误（str（e））值错误：索引。形状[-1]必须为

Answer 1

错误是由于序列长度变量的维数错误造成的。它必须是向量，而不是标量。

import tensorflow as tf

num_classes = 3
input_x = tf.placeholder(tf.int32, shape=[None, None], name="input_x")
input_y = tf.placeholder(tf.int32, shape=[None, sequence_length, num_classes])
sequence_length = tf.reduce_sum(tf.sign(input_x), 1)

# After some network operation you will come up with logits

logits = tf.placeholder(tf.float32, shape=[None, None, num_classes])
dense_y = tf.argmax(input_y, -1, output_type=tf.int32)
log_likelihood, _ = tf.contrib.crf.crf_log_likelihood(logits, dense_y, sequence_length