mean() received an invalid combination of arguments - got (Tensor, list, keepdim=bool), but expected one of: * (Tensor input) * (Tensor input, torch.dtype dtype) * (Tensor input, int dim, torch.dtype dtype, Tensor out) * (Tensor input, int dim, bool keepdim, torch.dtype dtype, Tensor out) * (Tensor input, int dim, bool keepdim, Tensor out)

[zhangxiaoyuyu]