1 reference to Concat
Microsoft.ML.TorchSharp (1)
Roberta\Modules\AttentionSelf.cs (1)
80
var newShape = DataUtils.
Concat
<long>(x.shape.AsSpan(0, x.shape.Length - 1), NumAttentionHeads, AttentionHeadSize);