TensorBatch
- class nvcv.TensorBatch
Container for a batch of tensors. The capacity of the container must be specified upfront in the batch initialization. The tensors in the batch may differ in shapes but they must have a uniform dimensionality, data type and layout.
- property capacity
Capacity of the tensor batch.
- clear(self: nvcv.TensorBatch) None
Remove all images from the TensorBatch.
- property dtype
Data type of tensors in the tensor batch. None if the batch is empty.
- property layout
Layout of the tensors in the tensor batch. None if the batch is empty.
- property ndim
Return the number of dimensions of the tensors or -1 for an empty batch
- popback(self: nvcv.TensorBatch, count: int = 1) None
Remove one or more images from the end of the TensorBatch.
- pushback(*args, **kwargs)
Overloaded function.
pushback(self: nvcv.TensorBatch, arg0: nvcv.Tensor) -> None
Add a new image to the end of the TensorBatch.
pushback(self: nvcv.TensorBatch, arg0: list[nvcv.Tensor]) -> None
Add multiple images to the end of the TensorBatch.
- nvcv.as_tensors(buffers: list[object] = [], layout: nvcv.TensorLayout | None = None) nvcv.TensorBatch
Wrap a list of external buffers as a batch of tensors, and tie the buffers lifetime to it