ExBurn.NifHelper (ex_burn v0.1.0)

Copy Markdown View Source

Helper module that wraps NIF calls and returns {:ok, result} tuples.

Summary

Functions

abs_tensor(a)
add_tensor(a, b)
broadcast_tensor(a, shape)
concat_tensor(a, b)
device_name()
div_tensor(a, b)
dot_tensor(a, b)
exp_tensor(a)
eye_tensor(size, type)
free_tensor(ref)
gpu_available()
iota_tensor(shape, axis, type)
layer_norm_tensor(a, dim, eps)
log_tensor(a)
matmul_tensor(a, b)
max_tensor(a)
max_tensor(a, axes)
mean_tensor(a)
mean_tensor(a, axes)
min_tensor(a)
min_tensor(a, axes)
mul_tensor(a, b)
neg_tensor(a)
new_tensor(data, 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)
sum_tensor(a, axes)
tanh_tensor(a)
tensor_shape(ref)
tensor_to_binary(ref)
to_cpu(tensor)
to_gpu(tensor)
transpose_tensor(a)
transpose_tensor(a, dim0, dim1)
zeros_tensor(shape, type)

Functions

abs_tensor(a)

add_tensor(a, b)

broadcast_tensor(a, shape)

concat_tensor(a, b)

device_name()

div_tensor(a, b)

dot_tensor(a, b)

exp_tensor(a)

eye_tensor(size, type)

free_tensor(ref)

gpu_available()

iota_tensor(shape, axis, type)

layer_norm_tensor(a, dim, eps)

log_tensor(a)

matmul_tensor(a, b)

max_tensor(a)

max_tensor(a, axes)

mean_tensor(a)

mean_tensor(a, axes)

min_tensor(a)

min_tensor(a, axes)

mul_tensor(a, b)

neg_tensor(a)

new_tensor(data, 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)

sum_tensor(a, axes)

tanh_tensor(a)

tensor_shape(ref)

tensor_to_binary(ref)

to_cpu(tensor)

to_gpu(tensor)

transpose_tensor(a)

transpose_tensor(a, dim0, dim1)

zeros_tensor(shape, type)