加入使用数组作为 shape 和 strides 创建 ninetoothed::Tensor 的方式

Author: voltjia

src/infiniop/ninetoothed/utils.h

@@ -19,6 +19,8 @@ class Tensor {
 
     Tensor(const void *data, std::initializer_list<Size> shape, std::initializer_list<Stride> strides) : Tensor{data, decltype(shape_){shape}, decltype(strides_){strides}} {}
 
+    Tensor(const void *data, const Size *shape, const Stride *strides, Size ndim) : data_{data}, shape_{shape, shape + ndim}, strides_{strides, strides + ndim}, ndim_{shape_.size()} {}
+
     Tensor(const T value) : value_{value}, data_{&value_}, ndim_{0} {}
 
     operator NineToothedTensor() { return {const_cast<Data>(data_), shape_.data(), strides_.data()}; }