Can anyone explain torch._assert with some examples , I am not able to understand from docs.Thanks in advance
torch._assert
should have the same functionality as the plain Python assert
method, but can be used in e.g. torch.fx
during tracing.
You could stick to the default assert
method unless you need to use it in some tracing methods.