ArgMax find both index of max value in s and max value.

Mean of a float32 vector.

Norm of a vector.

Normalize single vector according to: https://pytorch.org/docs/stable/generated/torch.nn.functional.normalize.html.

PathJoinSafe wrapper around filepath.Join to ensure that paths are correctly constructed if the path is a normal OS path, just use filepath.Join if the path is S3, trim any trailing slashes and construct it manually from the components so that double slashes (e.g.

SoftMax take a vector and calculate softmax scores of its values.