@override Tensor<Matrix> call(Tensor<dynamic> input) { return softmaxMatrix(input as Tensor<Matrix>); }