使用n维张量创建tf.train.Feature

import tensorflow as tf

def create_feature(data):
    # 将n维张量转换为适用的数据类型
    if data.dtype == tf.float32:
        data = tf.cast(data, tf.float32)
        feature = tf.train.Feature(float_list=tf.train.FloatList(value=data))
    elif data.dtype == tf.int64:
        data = tf.cast(data, tf.int64)
        feature = tf.train.Feature(int64_list=tf.train.Int64List(value=data))
    elif data.dtype == tf.string:
        data = tf.cast(data, tf.string)
        feature = tf.train.Feature(bytes_list=tf.train.BytesList(value=data))
    else:
        raise ValueError("Unsupported data type")
    
    return feature