(self, shape_1, shape_2)
| 159 | } |
| 160 | |
| 161 | def get_dict_example_1(self, shape_1, shape_2): |
| 162 | return { |
| 163 | "image": np.random.rand(*shape_1).astype("float32"), |
| 164 | "matrix": np.random.rand(*shape_2).astype("float32"), |
| 165 | "source": "bar", |
| 166 | } |
| 167 | |
| 168 | def get_dict_examples(self, shape_1, shape_2): |
| 169 | return { |