java形状_形状等于()

self.assertEqual(loss.shape, ())

以及导致它的选定行是:

BATCH_SIZE = 100

def dummy_input_fn():

image = tf.random_uniform([BATCH_SIZE, 784])

labels = tf.random_uniform([BATCH_SIZE, 1], maxval=9, dtype=tf.int32)

return image, labels

def make_estimator():

return tf.estimator.Estimator(

model_fn=mnist.model_fn, params={

'data_format': 'channels_last'

})

class Tests(tf.test.TestCase):

"""Run tests for MNIST model."""

def test_mnist(self):

classifier = make_estimator()

classifier.train(input_fn=dummy_input_fn, steps=2)

loss = eval_results['loss']

self.assertEqual(loss.shape, ())

t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])

tf.shape(t) # [2, 2, 3]

这两个打印对象形状的语句没有多大帮助:

print(loss.shape)

# prints `()`

print(tf.shape(loss))

# prints `Tensor("Shape:0", shape=(0,), dtype=int32)`

() 形状是什么意思?

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值