We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 90c1ae5 commit 4689eadCopy full SHA for 4689ead
README.md
@@ -1557,6 +1557,10 @@ input0 = pb_utils.Tensor.from_dlpack("INPUT0", pytorch_tensor)
1557
This method only supports contiguous Tensors that are in C-order. If the tensor
1558
is not C-order contiguous an exception will be raised.
1559
1560
+For python models with input or output tensors of type BFloat16 (BF16), the
1561
+`as_numpy()` method is not supported, and the `from_dlpack` and `to_dlpack`
1562
+methods must be used instead.
1563
+
1564
## `pb_utils.Tensor.is_cpu() -> bool`
1565
1566
This function can be used to check whether a tensor is placed in CPU or not.
0 commit comments