HeadTensor

constructor(reg: Tensor<FP32, Float>, cls: Tensor<FP32, Float>)(source)