Module elementwise 

Element-wise operations

Functions

add

Add two tensors

div

Divide two tensors

gelu

GELU activation

mul

Multiply two tensors

relu

ReLU activation

sigmoid

Sigmoid activation

softmax

Softmax

sub

Subtract two tensors