3×32×12×3var_0float32[3,3]Castcast_indices_0int32[2,1]input〈2×1〉GatherNDgathernd_0var_2float32[2,3]