在使用numpy的array时候,不能直接用del data[3]
这种写法,否则会出现标题错误。
需要用人家指定的numpy.delete(data, index, axis)
来删除。
torch中貌似没有直接删除列元素的函数,可以自己写:
def torch_delete(arr,index,axis):
if axis==0:
arr1 = arr[0:index]
arr2 = arr[index+1:]
elif axis==1:
arr1 = arr[:,:index]
arr2 = arr[:,index+1:]
return torch.cat((arr1,arr2),dim=axis)
使用方法和np的delete一样。