softmax method

Tensor softmax()

Implementation

Tensor softmax() {
  // Assuming your engine has a softmaxForward that handles [T, V] or [1, V]
  final h = engine.softmax_forward(_handle);
  return Tensor._raw(h, shape);
}