chickadee » nanograd » make-tensor32

Creates a 32-bit floating-point tensor with automatic differentiation support.

data

f32vector containing the tensor data

shape

list of dimensions, e.g., '(2 3) for a 2x3 matrix or '(10 2 3) for batch of 10 matrices

requires-grad?

whether to track gradients (default #t)

; Single vector
(define x (make-tensor32 (f32vector 1.0 2.0 3.0) '(3) requires-grad?: #t))

; Batch of vectors
(define batch (make-tensor32 (make-f32vector 60) '(10 6) requires-grad?: #t))