tensorflow2 中szie 和shape的区别

最近在根据pytorch的代码改写为tensorflow2的。其中,pytorch中求tensor的矩阵大小,直接用size就可以了。例如:

#简洁明了
>>> torch.empty(3, 4, 5).size()
torch.Size([3, 4, 5])

但是到TensorFlow2中,是不能这样了。TensorFlow2中应该使用shape

Returns the shape of a tensor.
tf.shape(
input, out_type=tf.dtypes.int32, name=None
)

t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.shape(t) 
<tf.Tensor: shape=(3,), dtype=int32, numpy=array([2, 2, 3], dtype=int32)>
tf.shape(t).numpy()
array([2, 2, 3], dtype=int32)
tf.size(t)  # 这个和torch.numel(t)是一个意思了
<tf.Tensor: shape=(), dtype=int32, numpy=12>

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值