Thanks a lot.
the output are not dict[str, tensor], but list[tensor] and the names are not right.
Do you have any suggestion with regard to this ?
Expect:
[{'xy': tensor([[0, 2],
        [4, 6]]), 'xz': tensor([[0, 2],
        [4, 6]]), 'yz': tensor([[0, 2],
        [4, 6]])}]
input_names=['0', '1', '2']
Actual:
[array([[0, 2],
       [4, 6]], dtype=int64), array([[0, 2],
       [4, 6]], dtype=int64), array([[0, 2],
       [4, 6]], dtype=int64)]