- a.device() a.gpu() a.numpy() a.ndim = ft.rank(a)
- aa = tf.convert_to_tensor(a)
- 数据类型转换:tf.cast(aa, dtype= tf.float32)
- TF数据格式:【b,h,w,c】 TH数据格式:【b,c,h,w】
- expandim()增加维度,squeen()减小维度 tf.tile()把单一维度复制n份
- ft1.0自定义网络模型需要继承 模型类 torch自定义数据集需要继承 数据集类
- pytorch 野马操作 mask = a<b ; a[mask] = b[mask] ; 会实现把a中比b中小的子集更新为b的子集
Tensorflow2.0 学习记录
最新推荐文章于 2024-10-15 10:20:47 发布