# DO NOT EDIT! This file was auto-generated by crates/build/re_types_builder/src/codegen/python/mod.rs
# Based on "crates/store/re_sdk_types/definitions/rerun/components/tensor_data.fbs".

# You can extend this class by creating a "TensorDataExt" class in "tensor_data_ext.py".

from __future__ import annotations

from .. import datatypes
from .._baseclasses import (
    ComponentBatchMixin,
    ComponentMixin,
)

__all__ = ["TensorData", "TensorDataBatch"]


class TensorData(datatypes.TensorData, ComponentMixin):
    """
    **Component**: An N-dimensional array of numbers.

    The number of dimensions and their respective lengths is specified by the `shape` field.
    The dimensions are ordered from outermost to innermost. For example, in the common case of
    a 2D RGB Image, the shape would be `[height, width, channel]`.

    These dimensions are combined with an index to look up values from the `buffer` field,
    which stores a contiguous array of typed values.
    """

    _BATCH_TYPE = None
    # You can define your own __init__ function as a member of TensorDataExt in tensor_data_ext.py

    # Note: there are no fields here because TensorData delegates to datatypes.TensorData


class TensorDataBatch(datatypes.TensorDataBatch, ComponentBatchMixin):
    _COMPONENT_TYPE: str = "rerun.components.TensorData"


# This is patched in late to avoid circular dependencies.
TensorData._BATCH_TYPE = TensorDataBatch  # type: ignore[assignment]