SetViewStrides
函数功能
设置aclTensor的ViewStride。
函数原型
SetViewStrides(const op::Strides &strides)
SetViewStrides(op::Strides &&strides)
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
strides |
输入 |
类型为FVector<int64_t>,表示aclTensor每一维的stride大小。 |
返回值说明
void
约束说明
无
调用示例
// 将input最后一维的stride扩大8倍。
void Func(const aclTensor *input) {
auto strides = input->GetViewStrides();
strides[strides.size() - 1] *= 8;
input->SetViewStrides(strides);
}
父主题: common_types