Get the shape of input array Parameters ---------- array : tvm.te.Tensor The source tensor. dtype : str, optional The target data type. Returns ------- result : tvm.te.Tensor The resulting tensor.
(array, dtype="int32")
| 703 | |
| 704 | |
| 705 | def shape(array, dtype="int32"): |
| 706 | """Get the shape of input array |
| 707 | |
| 708 | Parameters |
| 709 | ---------- |
| 710 | array : tvm.te.Tensor |
| 711 | The source tensor. |
| 712 | |
| 713 | dtype : str, optional |
| 714 | The target data type. |
| 715 | |
| 716 | Returns |
| 717 | ------- |
| 718 | result : tvm.te.Tensor |
| 719 | The resulting tensor. |
| 720 | """ |
| 721 | return cpp.shape(array, dtype) |
| 722 | |
| 723 | |
| 724 | def sequence_mask(data, valid_length, mask_value=0, axis=0): |
nothing calls this directly
no test coverage detected
searching dependent graphs…