2 创建一个形状为234的张量a并交换其第1个和第2个维度。打印结果张量的形状。tensorflow

• 日期: 2025-05-17
• 标签: 科技

import tensorflow as tf

a = tf.ones(shape=(2, 3, 4))
a = tf.transpose(a, perm=[1, 0, 2]) # 交换第1个和第2个维度
print(a.shape) # 输出 (3, 2, 4)

原文地址: https://www.cveoy.top/t/topic/eZhM 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录