Flatten
public final class Flatten : Model
Make the input tensor to be 1-D tensor (respecting N).
-
Declaration
Swift
public init(name: String = "")
-
Declaration
Swift
public func callAsFunction<T: DynamicGraph.TensorGroup>( _ input: T, streamContext: StreamContext? = nil ) -> T