I conda installed opt_einsum but hit the following error. Is there a specific version of opt_einsum that I should use? The installed one is opt_einsum-3.2.1.
Traceback (most recent call last):
  File "/private/home/shenli/local/miniconda/envs/torchdev/lib/python3.8/multiprocessing/pool.py", line 125, in worker
    result = (True, func(*args, **kwds))
  File "/private/home/shenli/local/miniconda/envs/torchdev/lib/python3.8/multiprocessing/pool.py", line 48, in mapstar
    return list(map(*args))
  File "/scratch/shenli/pytorch/test.py", line 229, in init_process
    return fn(rank, size)
  File "/scratch/shenli/pytorch/test.py", line 215, in run
    dogs.append(dog(img_tensor))
  File "/scratch/shenli/pytorch/torch/nn/modules/module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/scratch/shenli/pytorch/test.py", line 89, in forward
    f_gaussian_images = comp_mul(self.f_gaussian_pyramid, f_input)
  File "/scratch/shenli/pytorch/test.py", line 185, in comp_mul
    [op(x_real, y_real) - op(x_im, y_im), op(x_real, y_im) + op(x_im, y_real)],
  File "/private/home/shenli/local/miniconda/envs/torchdev/lib/python3.8/site-packages/opt_einsum/contract.py", line 473, in contract
    operands, contraction_list = contract_path(*operands,
  File "/private/home/shenli/local/miniconda/envs/torchdev/lib/python3.8/site-packages/opt_einsum/contract.py", line 222, in contract_path
    raise ValueError("Einstein sum subscript '{}' does not contain the "
ValueError: Einstein sum subscript 'buv' does not contain the correct number of indices for operand 1.