forward method

Tensor forward(Tensor input)

Implementation

Tensor forward(Tensor input) => input.softmax(dim);