class

CPU(T)

Inherits Num::Backend::Storage / Reference / Object

Constructors

new(shape : Array(Int), order : Num::OrderType, value : T)

Initialize a CPU storage from an initial capacity and an initial value, which will fill the buffer

Arguments

  • shape : Array(Int) - Shape of the parent Tensor
  • order : Array(Int) - Memory layout of the parent Tensor
  • value : T - Initial value to populate the buffer

Examples

CPU.new([10, 10], 3.4)
Source
new(shape : Array(Int), strides : Array(Int), value : T)

Initialize a CPU storage from an initial capacity and an initial value, which will fill the buffer

Arguments

  • shape : Array(Int) - Shape of the parent Tensor
  • strides : Array(Int) - Strides of the parent Tensor
  • value : T - Initial value to populate the buffer

Examples

CPU.new([10, 10], 3.4)
Source
new(hostptr : Pointer(T), shape : Array(Int), strides : Array(Int))

Initialize a CPU storage from a hostptr and initial shape. The shape is not required for this storage type, but is needed by other implementations to ensure copy requirements have the right pointer size.

Arguments

  • data : Pointer(T) - Existing databuffer for a Tensor
  • shape : Array(Int) - Shape of the parent Tensor
  • strides : Array(Int) - Strides of the parent Tensor

Examples

a = Pointer(Int32).malloc(10)
s = CPU.new(a, [5, 2])
Source
new(shape : Array(Int), order : Num::OrderType)

Initialize a CPU storage from an initial capacity. The data will be filled with zeros

Arguments

  • shape : Array(Int) - Shape of the parent Tensor
  • order : Array(Int) - Memory layout of the parent Tensor

Examples

CPU.new([2, 3, 4])
Source
new(shape : Array(Int), strides : Array(Int))

Initialize a CPU storage from an initial capacity. The data will be filled with zeros

Arguments

  • shape : Array(Int) - Shape of the parent Tensor
  • strides : Array(Int) - Strides of the parent Tensor

Examples

CPU.new([2, 3, 4])
Source

Class methods

base(dtype : U.class) : CPU(U).class forall U

Return a generic class of a specific generic type, to allow for explicit return types in functions that return a different storage type than the parent Tensor

Examples

a = CPU(Float32).new([10])

# Cannot do
# a.class.new ...

a.class.base(Float64).new([10])
Source

Instance methods

data

Raw Crystal pointer that holds a CPU(T)s data

Source
to_hostptr

Converts a CPU storage to a crystal pointer

Examples

a = CPU(Int32).new([3, 3, 2])
a.to_hostptr
Source
to_unsafe

Returns the raw Crystal pointer associated with a CPU(T)

Source