only batches of spatial targets supported (3D tensors) but got targets of dimension