Closed tzuhanchang closed 3 months ago
Use a custom softmax function, modified from torch_geometric.utils.softmax. This is a temporary fix to solve unconstrained value error raised by torch.onnx.dynamo_export.
softmax
torch_geometric.utils.softmax
torch.onnx.dynamo_export
Use a custom
softmax
function, modified fromtorch_geometric.utils.softmax
. This is a temporary fix to solve unconstrained value error raised bytorch.onnx.dynamo_export
.