例如下面的场景,如何对 t1 和 t2 下显然不同尺寸的 torch. ... trees = [TreeValue({ 'a': torch.randn(2, 4), 'b': torch.randn(3, 4), ...
確定! 回上一頁