多张量 axpby 的问题大小问题

csrc 目录中包含一个 cuda 中的 axpby 函数 (multi_tensor_axpby_kernel.cu)。我对问题的大小 (工作负载) 有个问题。不清楚该函数在 Apex 库之外是如何使用的。您是否知道张量列表 (如下所示) 中每个张量的典型维度以及张量列表的长度？谢谢。