提交 8bb2038d authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Remove TODO comment.

Nothing clever was going on, unknown dims were simply faked as having length 8 for JAX polymorphism export
上级 c4a3444b
......@@ -546,8 +546,6 @@ def einsum(subscripts: str, *operands: "TensorLike", optimize=None) -> TensorVar
"If you need this functionality open an issue in https://github.com/pymc-devs/pytensor/issues to let us know. "
)
# TODO: Is this doing something clever about unknown shapes?
# contract_path = _poly_einsum_handlers.get(ty, _default_poly_einsum_handler)
tensor_operands = [as_tensor(operand) for operand in operands]
shapes = [operand.type.shape for operand in tensor_operands]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论