1 write to SelfAttention
Microsoft.ML.TorchSharp (1)
NasBert\Modules\Layers\SelfAttentionLayer.cs (1)
32
SelfAttention
= new MultiHeadAttention(
2 references to SelfAttention
Microsoft.ML.TorchSharp (2)
NasBert\Modules\Layers\SelfAttentionLayer.cs (2)
64
var attention =
SelfAttention
.forward(query: x, key: x, value: x,
94
SelfAttention
.Dispose();