def f(x): x = x * 1 y = F.squeeze(x, [2, 3]) refs["x"] = TensorWeakRef(x) return y
def f(x): x = x * 1 y = F.squeeze(F.expand_dims(x, 2), 0) refs["x"] = TensorWeakRef(x) return y
def f(x): x = x * 1 y = F.expand_dims(x, [2, 3]) refs["x"] = TensorWeakRef(x) return y
def f(x): x = x * 1 y = x[[0, 2], [0, 2]] refs["x"] = TensorWeakRef(x) return y
def f(x): x = x * 1 y = x[1:-1, :2] refs["x"] = TensorWeakRef(x) return y
def f(x): x = x * 1 y = x.reshape(5, 2) refs["x"] = TensorWeakRef(x) return y
def f(x): x = x * 2 refs["x"] = TensorWeakRef(x) return relu(x)
def f(x, y): x = x * 2 refs["x"] = TensorWeakRef(x) refs["y"] = TensorWeakRef(y) return x + y