If your function takes other arguments that are not Tensors or Tensors that don't have requires_grad set, you can use a lambda to capture them.
確定! 回上一頁