FilesLast commitLast update
8 months ago
2 months ago
2 months ago
4 days ago
4 days ago
1 month ago
3 months ago
4 days ago
1 month ago
5 days ago
README

StridedSliceV2

English | 简体中文

Product Supported
Ascend 950PR&950DT products √
Atlas A3 products ×
Atlas A2 products ×
Atlas 200I/500 A2 inference products ×
Atlas inference products ×
Atlas training products ×

Function Description

  • Operator function: extracts a sub-tensor from the input tensor according to the specified start and end positions, axes, steps, and binary bit masks.

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
x Input Input tensor INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT, FLOAT16, BF16, BOOL, COMPLEX32, COMPLEX64, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, FLOAT8_E8M0 ND
begin Input Start value of each dimension INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT, FLOAT16, BF16, BOOL, COMPLEX32, COMPLEX64, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, FLOAT8_E8M0 ND
end Input End value of each dimension INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT, FLOAT16, BF16, BOOL, COMPLEX32, COMPLEX64, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, FLOAT8_E8M0 ND
axes Input Specifies the axes to be sliced, and the remaining axes are fully taken by default. INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT, FLOAT16, BF16, BOOL, COMPLEX32, COMPLEX64, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, FLOAT8_E8M0 ND
strides Input Interval at which values are taken at each point in each dimension INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT, FLOAT16, BF16, BOOL, COMPLEX32, COMPLEX64, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, FLOAT8_E8M0 ND
begin_mask Attribute This value specifies the dimensions whose begin values are ignored INT32, INT64 ND
end_mask Attribute This value specifies the dimensions whose end values are ignored INT32, INT64 ND
ellipsis_mask Attribute All dimensions starting from the specified dimension are selected until the user-specified begin is encountered INT32, INT64 ND
new_axis_mask Attribute Adds a shape with dimension 1 at the specified bit INT32, INT64 ND
shrink_axis_mask Attribute Forces the dimension at the corresponding index to be reduced to 1 INT32, INT64 ND
y Output Output tensor INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT, FLOAT16, BF16, BOOL, COMPLEX32, COMPLEX64, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN ND

Constraint Description

None

Invocation Description

Invocation Method Invocation Sample Description
Graph-mode invocation test_geir_strided_slice_v2 Invokes the StridedSliceV2 operator by building a graph through the operator IR