91db14020567987d8cda29a6140825c888e84917,thumt/data/record.py,,get_input_features,#Any#Any#Any#,105

Before Change

                     constant_values=bos_id)[:, :-1]

    // Final feature map.
    features = {
        "source": feature_map["inputs"],
        "target": targets,
        "source_length": tf.to_int32(
            tf.reduce_sum(
                tf.to_float(tf.not_equal(feature_map["inputs"], 0)),
                axis=1
            )
        ),
        "target_length": tf.to_int32(
            tf.reduce_sum(
                tf.to_float(tf.not_equal(feature_map["targets"], 0)),
                axis=1
            )
        ),
    }

    return features

After Change

        features = {
            "source": feature_map["source"],
            "target": feature_map["target"],
            "source_length": tf.squeeze(feature_map["source_length"], axis=1),
            "target_length": tf.squeeze(feature_map["target_length"], axis=1)
        }

Instances