@@ -35,6 +35,7 @@ using paddle::framework::LoDTensor;
35
35
using paddle::framework::LoD;
36
36
using paddle::framework::DDim;
37
37
using paddle::platform::CPUPlace;
38
+ using paddle::framework::make_ddim;
38
39
39
40
static void generatedata (const std::vector<std::string>& data,
40
41
const std::string& file_name) {
@@ -105,16 +106,21 @@ TEST(CTR_READER, read_data) {
105
106
106
107
std::vector<int64_t > label_value = {0 , 0 , 1 , 0 , 1 , 1 , 0 , 1 , 1 , 1 };
107
108
108
- std::vector<std::tuple<LoD, std::vector<int64_t >>> data_slot_6002{
109
- {{{0 , 1 , 2 , 7 }}, {0 , 0 , 10 , 11 , 12 , 13 , 14 }},
110
- {{{0 , 1 , 2 , 3 }}, {0 , 0 , 0 }},
111
- {{{0 , 1 , 2 , 3 }}, {30 , 0 , 40 }},
112
- {{{0 , 1 }}, {0 }}};
113
- std::vector<std::tuple<LoD, std::vector<int64_t >>> data_slot_6003{
114
- {{{0 , 1 , 4 , 5 }}, {1 , 5 , 6 , 7 , 0 }},
115
- {{{0 , 4 , 5 , 6 }}, {15 , 16 , 17 , 18 , 0 , 0 }},
116
- {{{0 , 1 , 3 , 4 }}, {31 , 35 , 36 , 41 }},
117
- {{{0 , 3 }}, {47 , 48 , 49 }}};
109
+ std::tuple<LoD, std::vector<int64_t >> a1 ({{0 , 1 , 2 , 7 }},
110
+ {0 , 0 , 10 , 11 , 12 , 13 , 14 });
111
+ std::tuple<LoD, std::vector<int64_t >> a2 ({{{0 , 1 , 2 , 3 }}, {0 , 0 , 0 }});
112
+ std::tuple<LoD, std::vector<int64_t >> a3 ({{{0 , 1 , 2 , 3 }}, {30 , 0 , 40 }});
113
+ std::tuple<LoD, std::vector<int64_t >> a4 ({{{0 , 1 }}, {0 }});
114
+ std::vector<std::tuple<LoD, std::vector<int64_t >>> data_slot_6002{a1, a2, a3,
115
+ a4};
116
+
117
+ std::tuple<LoD, std::vector<int64_t >> b1 ({{0 , 1 , 4 , 5 }}, {1 , 5 , 6 , 7 , 0 });
118
+ std::tuple<LoD, std::vector<int64_t >> b2 ({{0 , 4 , 5 , 6 }},
119
+ {15 , 16 , 17 , 18 , 0 , 0 });
120
+ std::tuple<LoD, std::vector<int64_t >> b3 ({{0 , 1 , 3 , 4 }}, {31 , 35 , 36 , 41 });
121
+ std::tuple<LoD, std::vector<int64_t >> b4 ({{0 , 3 }}, {47 , 48 , 49 });
122
+ std::vector<std::tuple<LoD, std::vector<int64_t >>> data_slot_6003{b1, b2, b3,
123
+ b4};
118
124
119
125
std::vector<DDim> label_dims = {{1 , 3 }, {1 , 3 }, {1 , 3 }, {1 , 1 }};
120
126
0 commit comments