Packages
Tensor library for Gleam/BEAM with a pure Gleam API, zero-copy views, and optional native acceleration
Current section
Files
Jump to
Current section
Files
include/viva_tensor@native@cuda_LinearLayer.hrl
-record(linear_layer, {
weight :: viva_tensor@native@cuda:accelerated_tensor(),
bias :: viva_tensor@native@cuda:accelerated_tensor(),
input_features :: integer(),
output_features :: integer(),
backend :: viva_tensor@native@cuda:acceleration_backend()
}).