Concat

public struct Concat<Element, Device> : LayerType, Codable where Element : NumericType, Device : DeviceType

Layer that concatenates a list of input tensors along their second dimension

  • Declaration

    Swift

    public var parameterPaths: [WritableKeyPath<`Self`, Tensor<Element, Device>>] { get }

  • Declaration

    Swift

    public var parameters: [Tensor<Element, Device>] { get }

  • Layer that concatenates a list of input tensors along their second dimension

    Declaration

    Swift

    public init()

  • Declaration

    Swift

    public func callAsFunction(_ inputs: [Tensor<Element, Device>]) -> Tensor<Element, Device>