RetinaHead 類別

定義

命名空間:

Microsoft.ML.TorchSharp.AutoFormerV2

組件:

Microsoft.ML.TorchSharp.dll

套件:

Microsoft.ML.TorchSharp v0.22.0-preview.24522.7

來源:

RetinaHead.cs

RetinaNet 的前端。

public class RetinaHead : TorchSharp.torch.nn.Module<System.Collections.Generic.List<TorchSharp.torch.Tensor>,(System.Collections.Generic.List<TorchSharp.torch.Tensor>, System.Collections.Generic.List<TorchSharp.torch.Tensor>)>

type RetinaHead = class
    inherit torch.nn.Module<List<torch.Tensor>, ValueTuple<List<torch.Tensor>, List<torch.Tensor>>>

Public Class RetinaHead
Inherits torch.nn.Module(Of List(Of torch.Tensor), ValueTuple(Of List(Of torch.Tensor), List(Of torch.Tensor)))

繼承

torch.nn.Module<List<TorchSharp.torch.Tensor>,ValueTuple<List<TorchSharp.torch.Tensor>,List<TorchSharp.torch.Tensor>>>

RetinaHead

建構函式

RetinaHead(Int32, Int32, Int32, Int32, Int32)

初始化 RetinaHead 類別的新執行個體。

方法

Dispose(Boolean)	RetinaNet 的前端。
forward(List<torch.Tensor>)	RetinaNet 的前端。