1{ 2 "datasetType": "TF", 3 "numRows": 5, 4 "columns": { 5 "next_sentence_labels": { 6 "type": "int64", 7 "rank": 1, 8 "shape": [1] 9 }, 10 "masked_lm_positions": { 11 "type": "int64", 12 "rank": 1, 13 "shape": [20] 14 }, 15 "masked_lm_ids": { 16 "type": "int64", 17 "rank": 1, 18 "shape": [20] 19 }, 20 "masked_lm_weights": { 21 "type": "float32", 22 "rank": 1, 23 "shape": [20] 24 }, 25 "input_mask": { 26 "type": "int64", 27 "rank": 1, 28 "shape": [128] 29 } 30 } 31} 32