T 9

作者: sumpig | 来源:发表于2019-02-05 11:06 被阅读0次

    class tf.data.Dataset

    • __init__
    __init__()
    
    • from_tensor_slices
    @staticmethod
    from_tensor_slices(tensors)
    
    • 示例
    input_data = [1, 2, 3, 4, 5]
    dataset = tf.data.Dataset.from_tensor_slices(input_data)
    
    • TextLineDataset
    • TFRecordDataset
    • make_one_shot_iterator
    • map
    • get_next
    def parser(record):
        features = ...
        return features['feat1'], features['feat2']
    
    dataset = tf.data.TFRecordDataset(input_files)    
    dataset = dataset.map(parser)
    iterator = dataset.make_one_shot_iterator()
    feat1, feat2 = iterator.get_next()
    

    相关文章

      网友评论

          本文标题:T 9

          本文链接:https://www.haomeiwen.com/subject/gxxvsqtx.html