Qwen-Image 20B中文图像生成微调工具包:含训练脚本、结构修复与推理优化
2026/6/7 12:00:53
tf.RaggedTensorSpec作用与参数说明tf.RaggedTensorSpec是 TensorFlow 中用于描述不规则张量(RaggedTensor)的“规格/签名”的类,常用来定义输入签名(如tf.function、SavedModel、Keras 输入等场景),告诉 TensorFlow 待处理的 RaggedTensor 应满足的形状、数据类型、不规则维度等约束。
spec = tf.RaggedTensorSpec(shape=[2, None, None], dtype=tf.int32, ragged_rank=2)shape=[2, None, None]
2(表示最外层维度有且仅有 2 个元素);None(表示这两个维度的长度是动态可变的,无固定值);ragged_rank=2,最终张量的“固定维度”是第 0 维(长度 2),第 1、2 维为不规则维度。dtype=tf.int32
1、5、100等)。ragged_rank=2
ragged_rank=2意味着:从第 1 维开始,连续 2 个维度(第 1、2 维)是不规则的(各元素的子维度长度可不同);# 外层固定2个元素,第1、2维长度可变rt=tf.ragged.constant([[[1,2],[3]],# 第0个元素:第1维长度2,第2维分别为2、1[[4],[5,6,7]]# 第1个元素:第1维长度2,第2维分别为1、3])该spec可用于:
tf.function的输入签名,约束传入的 RaggedTensor 必须匹配此规格;