当前位置：网站首页>tf.cast(),reduce_min(),reduce_max()

tf.cast(),reduce_min(),reduce_max()

2022-08-11 06:45:00 【phac123】

文章目录

简介

在tensorflow中可以使用tf.cast()对tensor变量的类型进行转换.使用tf.reduce_min和tf.reduce_max可以计算张量上元素的最大值和最小值.

实现

import tensorflow as tf

x1 = tf.constant([1., 2., 3.], dtype = tf.float64)
print("x1: ", x1)
x2 = tf.cast(x1, tf.int32)
print("x2: ", x2)
print("minimum of x2: ", tf.reduce_min(x2))
print("maxmum of x2: ", tf.reduce_max(x2))

在这里插入图片描述

原网站

版权声明
本文为[phac123]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_42596275/article/details/126254704

边栏推荐

猜你喜欢

随机推荐