Join a sequence of arrays along an existing axis. Parameters ---------- a_tuple : tuple of tvm.te.Tensor The arrays to concatenate axis : int, optional The axis along which the arrays will be joined. Default is 0. Returns ------- ret : tvm.te.Tensor
(a_tuple, axis=0)
| 393 | |
| 394 | |
| 395 | def concatenate(a_tuple, axis=0): |
| 396 | """Join a sequence of arrays along an existing axis. |
| 397 | |
| 398 | Parameters |
| 399 | ---------- |
| 400 | a_tuple : tuple of tvm.te.Tensor |
| 401 | The arrays to concatenate |
| 402 | |
| 403 | axis : int, optional |
| 404 | The axis along which the arrays will be joined. Default is 0. |
| 405 | |
| 406 | Returns |
| 407 | ------- |
| 408 | ret : tvm.te.Tensor |
| 409 | """ |
| 410 | return cpp.concatenate(a_tuple, axis) |
| 411 | |
| 412 | |
| 413 | def stack(tensors, axis=0): |
no test coverage detected
searching dependent graphs…