Hi,
Custom Functions are stateless. So defining the __init__() function does nothing. Only the forward/backward static functions are used.
If you create your Tensor directly inside the forward() function and save it on the ctx as ctx.a = a if you want to have it for the backward as well.