Skip to contents

Linear layer with a fixed binary mask on the weights. (Defined inside a function so the package loads without torch installed.)

Source: R/flows.R
masked_linear.Rd

Linear layer with a fixed binary mask on the weights. (Defined inside a function so the package loads without torch installed.)

Usage

masked_linear(in_features, out_features, mask)