-record(batch, {
    inputs :: viva_tensor@tensor:tensor(),
    targets :: viva_tensor@tensor:tensor()
}).