Boolean value of Tensor with more than one value is ambiguous