()
| 168 | |
| 169 | |
| 170 | def test_repeat_in_trace(): |
| 171 | @trace(symbolic=False) |
| 172 | def fun(data, repeats): |
| 173 | F.repeat(data, repeats) |
| 174 | |
| 175 | data = tensor(np.random.random([1, 2, 3]).astype(np.float32)) |
| 176 | |
| 177 | for i in range(1, 5): |
| 178 | repeats = tensor(i) |
| 179 | fun(data, repeats) |
| 180 | |
| 181 | |
| 182 | def test_print_in_trace(): |