# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/torch_core.ipynb.

# %% auto 0
__all__ = ['TensorCategory', 'TensorMultiCategory']

# %% ../nbs/torch_core.ipynb 1
# Contains code from:
# fastai - Apache License 2.0 - Copyright (c) 2023 fast.ai

# %% ../nbs/torch_core.ipynb 2
from fastai.torch_core import show_title
from fastai.torch_core import TensorCategory as _TensorCategory
from fastai.torch_core import TensorMultiCategory as _TensorMultiCategory

# %% ../nbs/torch_core.ipynb 4
class TensorCategory(_TensorCategory):
    "fastai's TensorCategory with a show method"
    def show(self, **kwargs):
        show_title(self.item(), **kwargs)

# %% ../nbs/torch_core.ipynb 5
class TensorMultiCategory(_TensorMultiCategory):
    "fastai's TensorMultiCategory with a show method"
    def show(self, **kwargs):
        show_title(self.item(), **kwargs)
