AutoFormerV2 建構函式

定義

命名空間:

Microsoft.ML.TorchSharp.AutoFormerV2

組件:

Microsoft.ML.TorchSharp.dll

套件:

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

來源:

AutoformerV2.cs

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

public AutoFormerV2(int numClasses, System.Collections.Generic.List<int> embedChannels, System.Collections.Generic.List<int> depths, System.Collections.Generic.List<int> numHeads, TorchSharp.torch.Device device = default);

new Microsoft.ML.TorchSharp.AutoFormerV2.AutoFormerV2 : int * System.Collections.Generic.List<int> * System.Collections.Generic.List<int> * System.Collections.Generic.List<int> * TorchSharp.torch.Device -> Microsoft.ML.TorchSharp.AutoFormerV2.AutoFormerV2

Public Sub New (numClasses As Integer, embedChannels As List(Of Integer), depths As List(Of Integer), numHeads As List(Of Integer), Optional device As torch.Device = Nothing)

參數

numClasses

Int32

物件類別目錄的數目。

embedChannels

List<Int32>

內嵌通道，可控制模型的規模。

depths

List<Int32>

區塊數目，可控制模型的縮放比例。

numHeads

List<Int32>

控制模型縮放比例的頭部數目。

device

TorchSharp.torch.Device

模型推斷所在的裝置。