invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number 解决办法https://github.com/NVIDIA/flownet2-pytorch/issues/113