I want to apply masked autoregressive flow to my architecture class ,but didn\'t know how to add it to my architecture . Does I need to implement a method for my class ? or