• notoraptor's avatar
    Add C code for CPU for tensor.Split. · ae3abc3d
    notoraptor 提交于
    Method: for each split, a view is created and then copied into an output.
    
    To come up:
    * Add C code for GPU.
    * Check if `self.len_splits` can be defined as a param op.
    ae3abc3d
basic.py 217.7 KB