I installed TensorComprehension with this command line following https://pytorch.org/2018/03/05/tensor-comprehensions.html: 4

conda install -c pytorch -c tensorcomp tensor_comprehensions

However, my pytorch version was reverted into 0.3.1.
Is TensorComprehension not supported yet for PyTorch 0.4 ?