假如我有一个local tensor维度为[128,256]，每行只有一个有效数据，其他数据为0。

我该如何配置SoftmaxFlashV2的入参从而对这个tensor求softmax flash呢？在SoftmaxFlashV2的接口接口里没看见指定输入数据有效数据个数的入参。