преобразование функции приведения тензорного потока с условием в pytorch | Kernel Crashes

Question

Я ищу эквивалент в Pytorch

binary_test = tf.cast(binary > 0.5, tf.float32)

Я пытался;

binary_test = binary.type(float32)
binary_test = torch.where(binary_test > 0.5, x, y)

где, x = Tensor([1]), y = Tensor([0])

При этом python ядро ​​вылетает.