AutoFormerV2 构造函数

参考

定义

命名空间:: Microsoft.ML.TorchSharp.AutoFormerV2

程序集:: Microsoft.ML.TorchSharp.dll

包:: Microsoft.ML.TorchSharp v0.21.1

重要

一些信息与预发行产品相关，相应产品在发行之前可能会进行重大修改。对于此处提供的信息，Microsoft 不作任何明示或暗示的担保。

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

模型推理的设备。

适用于