ExBurn.Nif (ex_burn v0.6.0)

Rustler NIF stubs for interfacing with the Burn deep learning framework.

The Rust side returns bare values for these operations. Fallible operations raise on failure (matching the pre-0.22 behaviour).

Summary

Functions

abs_tensor(a)

add_tensor(a, b)

backward_tensor(a)

broadcast_tensor(a, shape)

concat_tensor(a, b)

cross_entropy_loss(pred, target)

debug_nif_count()

Returns the number of NIF functions registered by the Rust library.

device_name()

div_tensor(a, b)

dot_tensor(a, b)

dropout(tensor, prob)

empty_tensor(shape, type)

exp_tensor(a)

eye_tensor(size, type)

free_tensor(tensor)

gpu_available()

grad_tensor(tensor, var)

iota_tensor(shape, axis, type)

layer_norm_tensor(a, dim, eps)

log_tensor(a)

matmul_tensor(a, b)

max_tensor(a)

mean_tensor(a)

min_tensor(a)

mse_loss(pred, target)

mul_tensor(a, b)

neg_tensor(a)

new_tensor(data, shape, type)

nif_abs_tensor(a)

nif_add_tensor(a, b)

nif_backward_tensor(a)

nif_broadcast_tensor(a, shape)

nif_concat_tensor(a, b)

nif_cross_entropy_loss(pred, target)

nif_device_name()

nif_div_tensor(a, b)

nif_dot_tensor(a, b)

nif_dropout(tensor, prob)

nif_empty_tensor(shape, type)

nif_exp_tensor(a)

nif_eye_tensor(size, type)

nif_free_tensor(tensor)

nif_gpu_available()

nif_grad_tensor(tensor, var)

nif_iota_tensor(shape, axis, type)

nif_layer_norm_tensor(a, dim, eps)

nif_log_tensor(a)

nif_matmul_tensor(a, b)

nif_max_tensor(a)

nif_mean_tensor(a)

nif_min_tensor(a)

nif_mse_loss(pred, target)

nif_mul_tensor(a, b)

nif_neg_tensor(a)

nif_new_tensor(data, shape, type)

nif_ones_tensor(shape, type)

nif_pow_tensor(a, exp)

nif_relu_tensor(a)

nif_reshape_tensor(a, shape)

nif_sigmoid_tensor(a)

nif_softmax_tensor(a, dim)

nif_sqrt_tensor(a)

nif_sub_tensor(a, b)

nif_sum_tensor(a)

nif_tanh_tensor(a)

nif_tensor_dtype(tensor)

nif_tensor_numel(tensor)

nif_tensor_shape(tensor)

nif_tensor_to_binary(tensor)

nif_to_cpu(tensor)

nif_to_gpu(tensor)

nif_transpose_tensor(a)

nif_zeros_tensor(shape, type)

ones_tensor(shape, type)

pow_tensor(a, exp)

relu_tensor(a)

reshape_tensor(a, shape)

sigmoid_tensor(a)

softmax_tensor(a, dim)

sqrt_tensor(a)

sub_tensor(a, b)

sum_tensor(a)

tanh_tensor(a)

tensor_dtype(tensor)

tensor_numel(tensor)

tensor_shape(tensor)

tensor_to_binary(tensor)

to_cpu(tensor)

to_gpu(tensor)

transpose_tensor(a)

zeros_tensor(shape, type)

Functions

abs_tensor(a)

add_tensor(a, b)

backward_tensor(a)

broadcast_tensor(a, shape)

concat_tensor(a, b)

cross_entropy_loss(pred, target)

debug_nif_count()

Returns the number of NIF functions registered by the Rust library.

device_name()

div_tensor(a, b)

dot_tensor(a, b)

dropout(tensor, prob)

empty_tensor(shape, type)

exp_tensor(a)

eye_tensor(size, type)

free_tensor(tensor)

gpu_available()

grad_tensor(tensor, var)

iota_tensor(shape, axis, type)

layer_norm_tensor(a, dim, eps)

log_tensor(a)

matmul_tensor(a, b)

max_tensor(a)

mean_tensor(a)

min_tensor(a)

mse_loss(pred, target)

mul_tensor(a, b)

neg_tensor(a)

new_tensor(data, shape, type)

nif_abs_tensor(a)

nif_add_tensor(a, b)

nif_backward_tensor(a)

nif_broadcast_tensor(a, shape)

nif_concat_tensor(a, b)

nif_cross_entropy_loss(pred, target)

nif_device_name()

nif_div_tensor(a, b)

nif_dot_tensor(a, b)

nif_dropout(tensor, prob)

nif_empty_tensor(shape, type)

nif_exp_tensor(a)

nif_eye_tensor(size, type)

nif_free_tensor(tensor)

nif_gpu_available()

nif_grad_tensor(tensor, var)

nif_iota_tensor(shape, axis, type)

nif_layer_norm_tensor(a, dim, eps)

nif_log_tensor(a)

nif_matmul_tensor(a, b)

nif_max_tensor(a)

nif_mean_tensor(a)

nif_min_tensor(a)

nif_mse_loss(pred, target)

nif_mul_tensor(a, b)

nif_neg_tensor(a)

nif_new_tensor(data, shape, type)

nif_ones_tensor(shape, type)

nif_pow_tensor(a, exp)

nif_relu_tensor(a)

nif_reshape_tensor(a, shape)

nif_sigmoid_tensor(a)

nif_softmax_tensor(a, dim)

nif_sqrt_tensor(a)

nif_sub_tensor(a, b)

nif_sum_tensor(a)

nif_tanh_tensor(a)

nif_tensor_dtype(tensor)

nif_tensor_numel(tensor)

nif_tensor_shape(tensor)

nif_tensor_to_binary(tensor)

nif_to_cpu(tensor)

nif_to_gpu(tensor)

nif_transpose_tensor(a)

nif_zeros_tensor(shape, type)

ones_tensor(shape, type)

pow_tensor(a, exp)

relu_tensor(a)

reshape_tensor(a, shape)

sigmoid_tensor(a)

softmax_tensor(a, dim)

sqrt_tensor(a)

sub_tensor(a, b)

sum_tensor(a)

tanh_tensor(a)

tensor_dtype(tensor)

tensor_numel(tensor)

tensor_shape(tensor)

tensor_to_binary(tensor)

to_cpu(tensor)

to_gpu(tensor)

transpose_tensor(a)

zeros_tensor(shape, type)