xbatcher.accessors module

xbatcher.accessors module#

class xbatcher.accessors.BatchAccessor(xarray_obj)[source]#

Bases: object

__init__(xarray_obj)[source]#: Batch accessor returning a BatchGenerator object via the generator method

generator(*args, **kwargs)[source]#

Return a BatchGenerator via the batch accessor

Parameters

*argsiterable: Positional arguments to pass to the BatchGenerator constructor.
**kwargsdict: Keyword arguments to pass to the BatchGenerator constructor.

class xbatcher.accessors.TFAccessor(xarray_obj)[source]#

Bases: object

__init__(xarray_obj)[source]#

to_tensor()[source]#: Convert this DataArray to a tensorflow.Tensor

class xbatcher.accessors.TorchAccessor(xarray_obj)[source]#

Bases: object

__init__(xarray_obj)[source]#

to_named_tensor()[source]#

Convert this DataArray to a torch.Tensor with named dimensions.

See https://pytorch.org/docs/stable/named_tensor.html

to_tensor()[source]#: Convert this DataArray to a torch.Tensor