TensorFlow入门--张量运算进阶

阅读 71

2022-02-09

第1关:在指定位置修改维度 - expand_dims方法与squeeze方法

本关任务:编写一个可以根据要求增加或减少张量维数的小程序。

# -*- coding: utf-8 -*-
import tensorflow as tf

# 根据action中参数的值修改tensor的形状
# tensor是一个TensorFlow中的张量
# action是一个TensorFlow中的一维张量 (向量

精彩评论(0)

0 0 举报