Hi,
If you don’t want any Tensor to be saved, just don’t save anything in the ctx and no Tensor will be saved for that Function.
Every Tensor is deleted as soon as it is not referenced by anything. So the input data will be deleted as soon as you don’t use it in your forward function.