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

Before Change

            )
        ),
        "target_length": tf.to_int32(
            tf.reduce_sum(
                tf.to_float(tf.not_equal(feature_map["targets"], 0)),
                axis=1
            )
        ),
    }

    return features

After Change

            "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)
        }

    return features

Instances