Tensor operation make [0.1,0.9] --> [0,1] (0.5 is the dividing line)

1

tensor operation
Suppose the value greater than 0.5 is set to 1, and the value less than 0.5 is set to 0.
for example，Tensor[0.1,0.2,0.8,0.9] --> Tensor[0,0,1,1].
I do not want to use for loop
so is there a corresponding function for the above operation?

2

tensor[tensor<0.5] = 0
tensor[tensor>=0.5] = 1

1 Like
3

Thank you very much!:smiley: