Theo chiều sâuConv2D

@frozen
public struct DepthwiseConv2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint

Lớp tích chập theo chiều sâu 2-D.

Lớp này tạo ra các bộ lọc tích chập có thể tách rời được tích hợp với đầu vào của lớp để tạo ra một tenxơ đầu ra.

  • Hạt nhân tích chập 4-D.

    Tuyên ngôn

    public var filter: Tensor<Scalar>

  • Vectơ thiên vị.

    Tuyên ngôn

    public var bias: Tensor<Scalar>

  • Chức năng kích hoạt theo phần tử.

    Tuyên ngôn

    @noDerivative
public let activation: Activation

  • Những bước tiến của cửa sổ trượt cho các kích thước không gian.

    Tuyên ngôn

    @noDerivative
public let strides: (Int, Int)

  • Thuật toán đệm cho tích chập.

    Tuyên ngôn

    @noDerivative
public let padding: Padding

  • Loại hàm kích hoạt theo phần tử.

    Tuyên ngôn

    public typealias Activation = @differentiable (Tensor<Scalar>) -> Tensor<Scalar>

  • Tạo lớp DepthwiseConv2D với bộ lọc, độ lệch, chức năng kích hoạt, bước tiến và phần đệm được chỉ định.

    Tuyên ngôn

    public init(
  filter: Tensor<Scalar>,
  bias: Tensor<Scalar>? = nil,
  activation: @escaping Activation = identity,
  strides: (Int, Int) = (1, 1),
  padding: Padding = .valid
)

    Thông số

    filter

    Hạt nhân tích chập 4-D.

    bias

    Vectơ thiên vị.

    activation

    Chức năng kích hoạt theo phần tử.

    strides

    Những bước tiến của cửa sổ trượt cho các kích thước không gian.

    padding

    Thuật toán đệm cho tích chập.

  • Trả về kết quả thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

    @differentiable
public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>

    Thông số

    input

    Đầu vào của lớp hình dạng, [số lô, chiều cao đầu vào, chiều rộng đầu vào, số kênh đầu vào]

    Giá trị trả về

    Đầu ra của hình dạng, [số lô, chiều cao đầu ra, chiều rộng đầu ra, số kênh đầu vào * hệ số nhân kênh]

  • Tạo lớp DepthwiseConv2D với hình dạng bộ lọc, bước tiến, phần đệm và chức năng kích hoạt theo phần tử được chỉ định.

    Tuyên ngôn

    public init(
  filterShape: (Int, Int, Int, Int),
  strides: (Int, Int) = (1, 1),
  padding: Padding = .valid,
  activation: @escaping Activation = identity,
  useBias: Bool = true,
  filterInitializer: ParameterInitializer<Scalar> = glorotUniform(),
  biasInitializer: ParameterInitializer<Scalar> = zeros()
)

    Thông số

    filterShape

    Hình dạng của hạt tích chập 4-D có dạng, [chiều rộng bộ lọc, chiều cao bộ lọc, số kênh đầu vào, hệ số nhân kênh].

    strides

    Những bước tiến của cửa sổ trượt cho các chiều không gian/không gian-thời gian.

    padding

    Thuật toán đệm cho tích chập.

    activation

    Chức năng kích hoạt theo phần tử.

    filterInitializer

    Trình khởi tạo để sử dụng cho các tham số bộ lọc.

    biasInitializer

    Trình khởi tạo để sử dụng cho các tham số sai lệch.