util

func ArgMax ¶

func ArgMax(s []float32) (int, float32, error)

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

func CloseFile ¶

func CloseFile(file io.Closer) error

func GetPathType ¶

func GetPathType(path string) string

func Mean ¶

func Mean(vector []float32) float32

Mean of a float32 vector.

func Norm ¶ added in v0.1.1

func Norm(v []float32, p int) float64

Norm of a vector.

func Normalize ¶ added in v0.1.1

func Normalize(embedding []float32, p int) []float32

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

func PathJoinSafe ¶

func PathJoinSafe(elem ...string) string

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. s3://) are preserved.

func ReadFileBytes ¶

func ReadFileBytes(filename string) ([]byte, error)

func Sigmoid ¶ added in v0.0.9

func Sigmoid(s []float32) []float32

func SoftMax ¶

func SoftMax(vector []float32) []float32

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

func SumSlice ¶

func SumSlice(s []float64) float64