Attention 类

定义

命名空间:

Microsoft.ML.TorchSharp.AutoFormerV2

程序集:

Microsoft.ML.TorchSharp.dll

包:

Microsoft.ML.TorchSharp v0.23.0-preview.1.25125.4

Source:

Attention.cs

注意层。

public class Attention : TorchSharp.torch.nn.Module<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>

type Attention = class
    inherit torch.nn.Module<torch.Tensor, torch.Tensor, torch.Tensor>

Public Class Attention
Inherits torch.nn.Module(Of torch.Tensor, torch.Tensor, torch.Tensor)

继承

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

Attention

构造函数

Attention(Int32, Int32, Int32, Int32, List<Int32>)

初始化 Attention 类的新实例。

方法

Dispose(Boolean)	注意层。
forward(torch+Tensor, torch+Tensor)	注意层。