Try using (depth,) as the shape on the “reshape”. The point is it needs to be a 1D tensor, not a 0D tensor.