oneflow.Tensor.grad_fn

Tensor.grad_fn

Return the function that created this tensor if it’s requires_grad is True.