2023年11月10日
矩阵维度变换--einops库
摘要： import einops # 创建一个形状为(batch_size, seq_length, hidden_dim)的张量 tensor = tf.constant([[[1, 2, 3, 4], [5, 6, 7, 8]], [[9, 10, 11, 12], [13, 14, 15, 16]] 阅读全文
posted @ 2023-11-10 15:49 山…隹 阅读(35) 评论(0) 推荐(0) 编辑