語言

AutoFormerV2 建構函式

定義

初始化 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

模型推論的裝置。

適用於