attributeerror 'tuple' object has no attribute 'dim'