def test3d(self): input_tensor = constant_op.constant([[[1, 2]], [[3, 4]]]) expected_output_array = [[[1, 2], [0, 0]], [[3, 4], [0, 0]]] op = loss_functions.insert_slice_in_zeros(input_tensor, 1, 2, 0) with self.test_session() as sess: actual_output_array = sess.run(op) self.assertAllEqual(expected_output_array, actual_output_array)
def testBadShape(self): bad_shaped_ones = array_ops.ones(shape=[1, 3]) # n.b. shape[1] != 1 with self.assertRaises(ValueError): loss_functions.insert_slice_in_zeros(bad_shaped_ones, 1, 42, 17)