| x |
输入 |
输入的张量 |
INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN |
ND |
| begin |
输入 |
每个维度的起始值 |
INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN |
ND |
| end |
输入 |
每个维度的结束值 |
INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN |
ND |
| strides |
输入 |
每个维度上每个点取值的跨度 |
INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN |
ND |
| begin_mask |
属性 |
这个值指定哪些维度的begin被忽略 |
INT32、INT64 |
ND |
| end_mask |
属性 |
这个值指定哪些维度的end被忽略 |
INT32、INT64 |
ND |
| ellipsis_mask |
属性 |
从指定的维度开始全选,直到遇到用户指定begin才退出 |
INT32、INT64 |
ND |
| new_axis_mask |
属性 |
在指定位上增加维度为1的shape |
INT32、INT64 |
ND |
| shrink_axis_mask |
属性 |
把对应索引处维度强制降为1 |
INT32、INT64 |
ND |
| y |
输出 |
输出张量 |
INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN |
ND |