Documentation ¶
Overview ¶
Package types is mostly a top level directory for GoMLX important types. See sub-packages `shapes`, `tensor` and `slices`.
This package also provides the types: Set.
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Set ¶
type Set[T comparable] map[T]struct{}
Set implements a Set for the key type T.
func MakeSet ¶
func MakeSet[T comparable](size ...int) Set[T]
MakeSet returns an empty Set of the given type. Size is optional, and if given will reserve the expected size.
Directories ¶
Path | Synopsis |
---|---|
Package exceptions provides helper functions to leverage Go's `panic`, `recover` and `defer` as an "exceptions" system.
|
Package exceptions provides helper functions to leverage Go's `panic`, `recover` and `defer` as an "exceptions" system. |
Package keepalive provides a simple Acquire and Release mechanism to make sure data is kept alive in between.
|
Package keepalive provides a simple Acquire and Release mechanism to make sure data is kept alive in between. |
Package shapes defines Shape and DType and associated tools.
|
Package shapes defines Shape and DType and associated tools. |
Package slices are
|
Package slices are |
Package tensor provides a `Tensor` interface with 2 different implementations: `Local` and `Device`, they differ on where their values are stored: in the local (host) CPU, or on an accelerator device (TPU, GPU, but could be also the CPU if no accelerators are available).
|
Package tensor provides a `Tensor` interface with 2 different implementations: `Local` and `Device`, they differ on where their values are stored: in the local (host) CPU, or on an accelerator device (TPU, GPU, but could be also the CPU if no accelerators are available). |
image
Package image provides several functions to transform images back and forth from tensors.
|
Package image provides several functions to transform images back and forth from tensors. |
Click to show internal directories.
Click to hide internal directories.