discopat.nn_training.detr.nested_tensor

Functions

nested_tensor_from_tensor_list(tensor_list)

Classes

NestedTensor(tensors, mask)

class discopat.nn_training.detr.nested_tensor.NestedTensor(tensors, mask)[source]

Bases: object

Parameters:

mask (Tensor | None)

decompose()[source]
to(device)[source]
Parameters:

device (Device)

Return type:

NestedTensor

discopat.nn_training.detr.nested_tensor.nested_tensor_from_tensor_list(tensor_list)[source]
Parameters:

tensor_list (list[Tensor])