tech.compute.tensor — techascent/tech.compute 4.24-1

Functions for dealing with tensors with the compute system

clone-to-device clj

(clone-to-device input-tens)

(clone-to-device input-tens options)

Clone a host tensor to a device. Tensor must have relatively straighforward dimensions (transpose OK but arbitrary reorder or offset not OK) or :force? must be specified. options: :force? Copy tensor to another buffer if necessary. :sync? Sync with stream to ensure copy operation is finished before moving forward.

clone-to-host clj

(clone-to-host device-tens)

(clone-to-host device-tens options)

ensure-device clj

(ensure-device input-tens)

(ensure-device input-tens options)

Ensure a tensor can be used on a device. Some devices can use CPU tensors.

ensure-host clj

(ensure-host device-tens)

(ensure-host device-tens options)

new-device-tensor clj

(new-device-tensor shape)

(new-device-tensor shape options)

new-host-tensor clj

(new-host-tensor shape)

(new-host-tensor shape options)