-record(size_mismatch, {
    axis :: viva_tensor@axis:axis(),
    expected :: integer(),
    got :: integer()
}).