Sự giãn nở2D

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

Lớp giãn nở hình thái 2-D

Lớp này trả về độ giãn hình thái của tensor đầu vào với các bộ lọc được cung cấp

  • Bộ lọc giãn nở 4-D.

    Tuyên ngôn

    public var filter: Tensor<Scalar>
  • 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 sự giãn nở.

    Tuyên ngôn

    @noDerivative
    public let padding: Padding
  • Hệ số giãn nở cho kích thước không gian.

    Tuyên ngôn

    @noDerivative
    public let rates: (Int, Int)
  • Tạo lớp Dilation2D với bộ lọc, bước tiến, độ giãn và phần đệm được chỉ định.

    Tuyên ngôn

    public init(
      filter: Tensor<Scalar>,
      strides: (Int, Int) = (1, 1),
      rates: (Int, Int) = (1, 1),
      padding: Padding = .valid
    )

    Thông số

    filter

    Bộ lọc giãn nở 4-D có hình dạng [chiều cao bộ lọc, chiều rộng bộ lọc, số kênh đầu vào, số lượng kênh đầu ra].

    strides

    Các bước của cửa sổ trượt cho các kích thước không gian, tức là (chiều cao bước, chiều rộng bước).

    rates

    Tốc độ giãn nở cho các kích thước không gian, tức là (chiều cao giãn nở, chiều rộng giãn nở).

    padding

    Thuật toán đệm cho sự giãn nở.

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

    Kích thước không gian đầu ra được tính như sau:

    chiều cao đầu ra = (chiều cao đầu vào + 2 * chiều cao đệm - (chiều cao giãn nở * (chiều cao bộ lọc - 1) + 1)) / chiều cao sải chân + 1

    chiều rộng đầu ra = (chiều rộng đầu vào + 2 * chiều rộng đệm - (chiều rộng giãn nở * (chiều rộng bộ lọc - 1) + 1)) / chiều rộng sải chân + 1

    và kích thước phần đệm được xác định bởi sơ đồ phần đệm.

    Ghi chú

    Kích thước đệm bằng 0 khi sử dụng .valid .

    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 [kích thước lô, chiều cao đầu vào, chiều rộng đầu vào, số lượng 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 ra].