Torch Core

Basic PyTorch functions used in fastxtend

source

TensorCategory

 TensorCategory (x, **kwargs)

fastai’s TensorCategory with a show method


source

TensorMultiCategory

 TensorMultiCategory (x, **kwargs)

fastai’s TensorMultiCategory with a show method