Versions in this module Expand all Collapse all v1 v1.1.1 Jul 5, 2021 v1.1.0 Jul 3, 2021 Changes in this version + var CPU Device = Device + var CUDA Cuda = Cuda + func DTypeSize(dt DType) (retVal uint, err error) + func ElementGoType(data interface{}) (retVal reflect.Type, err error) + func IsSupportedScalar(k reflect.Kind) bool + func ToGoType(dtype DType) (retVal reflect.Type, err error) + func TypeOf(dt DType, shape []int64) (retVal reflect.Type, err error) + type CInt = int32 + func DType2CInt(dt DType) (retVal CInt, err error) + type Cuda Device + func (cu Cuda) CudnnIsAvailable() bool + func (cu Cuda) CudnnSetBenchmark(b bool) + func (cu Cuda) DeviceCount() int64 + func (cu Cuda) IsAvailable() bool + type DType struct + var Bool DType = DType{ ... } + var Double DType = DType{ ... } + var Float DType = DType{ ... } + var Int DType = DType{ ... } + var Int16 DType = DType{ ... } + var Int64 DType = DType{ ... } + var Int8 DType = DType{ ... } + var Uint8 DType = DType{ ... } + func CInt2DType(v CInt) (dtype DType, err error) + func DTypeFromData(data interface{}) (retVal DType, err error) + func DataDType(v interface{}, shape []int64) (retVal DType, err error) + func ElementDType(v interface{}) (retVal DType, err error) + func ToDType(typ reflect.Type) (retVal DType, err error) + func (dt DType) CInt() (retVal CInt) + type DTypeDevice struct + DType DType + Device Device + var DoubleCPU DTypeDevice = DTypeDevice{ ... } + var DoubleCUDA DTypeDevice = DTypeDevice{ ... } + var FloatCPU DTypeDevice = DTypeDevice{ ... } + var FloatCUDA DTypeDevice = DTypeDevice{ ... } + var Int64CPU DTypeDevice = DTypeDevice{ ... } + var Int64CUDA DTypeDevice = DTypeDevice{ ... } + type Device struct + Name string + Value int + func CudaBuilder(v uint) Device + func CudaIfAvailable() Device + func NewCuda() Device + func (d Device) CInt() CInt + func (d Device) CudaIfAvailable() Device + func (d Device) IsCuda() bool + func (d Device) OfCInt(v CInt) Device