MaskedFill

public final class MaskedFill : Model

The masked fill model. If the value equal to a given constant, fill with another constant.


                    
                    
                    init(equalTo:fillWith:name:)

Declaration

Swift

public init(equalTo: Float, fillWith: Float, name: String = "")


                    
                    
                    callAsFunction(_:_:streamContext:)

Declaration

Swift

public func callAsFunction<T: DynamicGraph.TensorGroup, U: DynamicGraph.TensorGroup>(
  _ left: T, _ right: U, streamContext: StreamContext? = nil
) -> T