RetinaHead 类

定义

命名空间:

Microsoft.ML.TorchSharp.AutoFormerV2

程序集:

Microsoft.ML.TorchSharp.dll

包:

Microsoft.ML.TorchSharp v0.21.1

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 类的新实例。

方法

forward(List<torch.Tensor>)

RetinaNet 的负责人。