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