การขยาย2D

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

ชั้นการขยายตัวทางสัณฐานวิทยา 2 มิติ

เลเยอร์นี้ส่งคืนการขยายทางสัณฐานวิทยาของเทนเซอร์อินพุตด้วยตัวกรองที่ให้มา

  • ตัวกรองการขยาย 4-D

    คำประกาศ

    public var filter: Tensor<Scalar>
  • ความก้าวหน้าของหน้าต่างบานเลื่อนเพื่อมิติเชิงพื้นที่

    คำประกาศ

    @noDerivative
    public let strides: (Int, Int)
  • อัลกอริธึมการขยายสำหรับการขยาย

    คำประกาศ

    @noDerivative
    public let padding: Padding
  • ปัจจัยการขยายสำหรับมิติเชิงพื้นที่

    คำประกาศ

    @noDerivative
    public let rates: (Int, Int)
  • สร้างเลเยอร์ Dilation2D พร้อมด้วยตัวกรอง การก้าวย่าง การขยายและการเติมที่ระบุ

    คำประกาศ

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

    พารามิเตอร์

    filter

    ตัวกรองการขยาย 4 มิติของรูปร่าง [ความสูงของตัวกรอง ความกว้างของตัวกรอง จำนวนช่องอินพุต จำนวนช่องเอาต์พุต]

    strides

    ก้าวของหน้าต่างบานเลื่อนสำหรับมิติเชิงพื้นที่ เช่น (ความสูงก้าว ความกว้างก้าว)

    rates

    อัตราการขยายสำหรับมิติเชิงพื้นที่ เช่น (ความสูงของการขยาย ความกว้างของการขยาย)

    padding

    อัลกอริธึมการขยายสำหรับการขยาย

  • ส่งกลับผลลัพธ์ที่ได้จากการใช้เลเยอร์กับอินพุตที่กำหนด

    มิติเชิงพื้นที่เอาต์พุตคำนวณได้ดังนี้:

    ความสูงเอาท์พุต = (ความสูงอินพุต + 2 * ความสูงของการเสริม - (ความสูงการขยาย * (ความสูงของตัวกรอง - 1) + 1)) / ความสูงก้าวย่าง + 1

    ความกว้างเอาต์พุต = (ความกว้างอินพุต + 2 * ความกว้างของช่องว่างภายใน - (ความกว้างของการขยาย * (ความกว้างของตัวกรอง - 1) + 1)) / ความกว้างก้าว + 1

    และขนาดการเสริมจะถูกกำหนดโดยโครงร่างการเสริม

    บันทึก

    ขนาดช่องว่างภายในเท่ากับศูนย์เมื่อใช้ .valid

    คำประกาศ

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

    พารามิเตอร์

    input

    การป้อนข้อมูลไปยังเลเยอร์ของรูปร่าง [ขนาดแบทช์ ความสูงของอินพุต ความกว้างอินพุต จำนวนช่องอินพุต]

    ค่าส่งคืน

    ผลลัพธ์ของรูปร่าง [จำนวนแบทช์ ความสูงเอาต์พุต ความกว้างเอาต์พุต จำนวนช่องสัญญาณเอาต์พุต]