【深度学习】_amax() got an unexpected keyword argument ‘dim‘ 解决方案

在定义一个点云数据pc后，想使用pc.max（dim=0），然后出现了_amax() got an unexpected keyword argument ‘dim‘

这个是因为对于tensor类型的数据和ndarray类型的数据都有一个max/min函数，tensor后面的max函数是torch自带的，后者是numpy定义的。

如果pc是tensor类型的数据才能使用pc.max（dim=0），如果是ndarray类型，就要使用pc.max（axis=0）。

显然上面报错的原因是pc其实是ndarray数组，但是使用了来自tensor的max函数。这里可以通过pc = torch.from_numpy(pc)将其从ndarray转化成rensor数据，或者将dim改成axis即可。