We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 90052a8 commit 461bd2cCopy full SHA for 461bd2c
aten/src/ATen/NestedTensorImpl.cpp
@@ -177,7 +177,9 @@ NestedTensorImpl::NestedTensorImpl(
177
C10_LOG_API_USAGE_ONCE("torch.NestedTensor");
178
TORCH_WARN_ONCE(
179
"The PyTorch API of nested tensors is in prototype stage and will change "
180
- "in the near future.");
+ "in the near future. We recommend specifying layout=torch.jagged when constructing "
181
+ "a nested tensor, as this layout receives active development, has better operator "
182
+ "coverage, and works with torch.compile.");
183
auto storage_device = storage_.device();
184
TORCH_INTERNAL_ASSERT(
185
storage_device.is_cpu() || storage_device.is_cuda() || storage_device.is_xpu() || storage_device.is_privateuseone(),
0 commit comments