Ejemplo n.º 1
0
def test_item(t: Tensor) -> float:
    t = t.sum()
    return t.item()
Ejemplo n.º 2
0
 def f(x: Tensor) -> Tuple[Tensor, Tuple[Tensor, Tensor]]:
     x = x.square()
     return x.sum(), (x, x + 1)